Advertisement
deushiro

Untitled

Jan 20th, 2020
189
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.82 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3. #include <cmath>
  4. #include <algorithm>
  5.  
  6. using namespace std;
  7.  
  8. typedef long long ll;
  9.  
  10. int power(int v, int u) {
  11.     if (u == 0  ) {
  12.         return 1;
  13.     }
  14.     if (u % 2 == 0){
  15.         return power(v, u / 2) * power(v, u / 2);
  16.     }
  17.     else {
  18.         return v * power(v, u - 1);
  19.     }
  20. }
  21.  
  22.  
  23. int main()
  24. {
  25.     ios_base::sync_with_stdio(false);
  26.     cin.tie(0);
  27.     cout.tie(0);
  28.     int n, k;
  29.     cin >> n >> k;
  30.     vector<int> dp(n + 1);
  31.     if (k == 1) {
  32.         cout << 1 << "\n";
  33.         return 0;
  34.     }
  35.     for (int i = 0; i < k; ++i) {
  36.         dp[i] = power(2, i);
  37.     }
  38.     for (int i = 0; i < k; ++i) {
  39.         dp[k] += dp[i];
  40.     }
  41.     for (int i = k + 1; i <= n; ++i) {
  42.         dp[i] = dp[i - 1] + dp[i - 1] - dp[i - k - 1];
  43.     }
  44.     cout << dp[n] << "\n";
  45. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement