Advertisement
Josif_tepe

Untitled

Feb 18th, 2022
761
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.25 KB | None | 0 0
  1. #include <iostream>
  2. #include <algorithm>
  3. #include <vector>
  4. #include <cstring>
  5. using namespace std;
  6. int n,k;
  7. const int mod = 1e9 + 7;
  8. int a[105];
  9. int memo[100005];
  10. void add_mod(int &a, int b) {
  11.     a += b;
  12.     if(a >= mod) {
  13.         a -= mod;
  14.     }
  15. }
  16. void subtract_mod(int &a, int b) {
  17.     a -= b;
  18.     if(a < 0) {
  19.         a += mod;
  20.     }
  21. }
  22. int main() {
  23.     cin >> n >> k;
  24.     for (int i = 0; i < n; i++) {
  25.         cin >> a[i];
  26.     }
  27.     memset(memo,0,sizeof(memo));
  28.     memo[0] = 1; // memo[i] - broj na nacini da stigneme so i candies
  29.     for(int at = 0; at < n; at++) {
  30.         vector<int> pref_sum(k + 5, 0);
  31.         for(int chocolates_left = k; chocolates_left >= 0; chocolates_left--) {
  32.             int start = chocolates_left + 1;
  33.             int end = chocolates_left + min(a[at], k - chocolates_left);
  34.             if(start <= end) {
  35.                 add_mod(pref_sum[start], memo[chocolates_left]);
  36.                 subtract_mod(pref_sum[end + 1], memo[chocolates_left]);
  37.             }
  38.            
  39.         }
  40.         int sum = 0;
  41.         for(int j = 0; j <= k; j++) {
  42.             add_mod(sum, pref_sum[j]);
  43.             add_mod(memo[j], sum);
  44.         }
  45.     }
  46.    
  47.         cout << memo[k] << endl;
  48.     return 0;
  49. }
  50. // min(1, )
  51.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement