Editing 0730 - Minus K (section)

==Rezolvare==
<syntaxhighlight lang="python3" line="1">
MOD = 2011

def numar_siruri(N, K):

   # Inițializăm matricea dp cu 0-uri
   dp = [[0] * 2 for _ in range(N + 1)]
   # Prima coloană este 1, pentru că există un singur șir de lungime 0, și anume șirul vid
   dp[0][0] = 1
   # Calculăm numărul de șiruri
   for i in range(1, N + 1):
       # Dacă adăugăm + în șirul anterior, nu adăugăm un semn - pe poziția i
       dp[i][0] = (dp[i - 1][0] + dp[i - 1][1]) % MOD
       # Dacă adăugăm - în șirul anterior, verificăm dacă am adăugat deja K semne - consecutive
       if i >= K:
           dp[i][1] = (dp[i - 1][0] - dp[i - K][1] + MOD) % MOD
       else:
           dp[i][1] = dp[i - 1][0]
   return (dp[N][0] + dp[N][1]) % MOD

if __name__ == "__main__":

   with open("minusk.in", "r") as f:
       N, K = map(int, f.readline().split())
   rezultat = numar_siruri(N, K)
   with open("minusk.out", "w") as g:
       g.write(str(rezultat) + "\n")
</syntaxhighlight>