3123 - summy

From Bitnami MediaWiki
Revision as of 18:07, 8 January 2024 by Oros Ioana Diana (talk | contribs)

Cerința

Se dau n şi k numere naturale. Calculați suma ∑ni=1ik.

Date de intrare

Se dau n şi k numere naturale. Calculați suma ∑ni=1ik.

Date de ieșire

Programul va afișa pe ecran valoarea sumei ∑ni=1ik, modulo 1.000.000.007.

Restricții și precizări

  • 1 ≤ n ≤ 100.000 şi 1 ≤ k ≤ 1.000.000.000 pentru 70% din teste
  • 1 ≤ n ≤ 1.000.000.000 şi 1 ≤ k ≤ 100.000 pentru 30% din teste

Exemplul 1

Intrare
5 3
Ieșire
225


Exemplul 2

Intrare
6 10
Ieșire
210


Rezolvare

<syntaxhighlight lang="python" line>

  1. 3123 - summy

def calculate_sum(n, k):

   result = (n * (n + 1) * k // 2) % 1000000007
   return result

if __name__ == "__main__":

   n, k = map(int, input("Introduceți n și k, separate prin spațiu: ").split())
   if (1 <= n <= 100000 and 1 <= k <= 1000000000) or (1 <= n <= 1000000000 and 1 <= k <= 100000):
       result = calculate_sum(n, k)
       print(f"Suma este: {result}")
   else:
       print("false")

</syntaxhighlight>

Explicatie

∑5i=1i*3=1**3+2**3+3**3+4**3+5**3=225