#include<bits/stdc++.h> using namespace std; const int MOD = 1e9 + 7; struct para { int a,b; }; bool operator<(para a,para b) { return a.a < b.a; } int main() { ios_base::sync_with_stdio(0); int n; cin >> n; vector<int> vec(n); // int max_ai = 0; for(int k=0;k<n;k++) { cin >> vec[k]; //max_ai = max(max_ai,vec[k]); } sort(vec.begin(),vec.end()); vector<int> tab(5001,0); tab[0] = 1; int wynik =0; int ile = 0; for(int k=0;k<n;k++) {// // cout << "LOW"<<endl; vector<para> add; int il = 0; for(int i=vec[k]-1;i<=5000;i++) {// cout << *x <<endl; if(tab[i]>0){ wynik+=tab[i]; wynik%=MOD; if(i + vec[k] <= 5000) add.push_back({i + vec[k],tab[i]}); else il+=tab[i]; il%=MOD;} } wynik+=ile; wynik%=MOD; ile*=2; ile%=MOD; ile+=il; ile%=MOD; for(auto i: add) { tab[i.a] +=i.b; tab[i.a]%=MOD; } //cout << "WTF"<<endl; } //for(auto t:secik) // cout << t.a << " " << t.b<<endl; cout << wynik<<endl; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | #include<bits/stdc++.h> using namespace std; const int MOD = 1e9 + 7; struct para { int a,b; }; bool operator<(para a,para b) { return a.a < b.a; } int main() { ios_base::sync_with_stdio(0); int n; cin >> n; vector<int> vec(n); // int max_ai = 0; for(int k=0;k<n;k++) { cin >> vec[k]; //max_ai = max(max_ai,vec[k]); } sort(vec.begin(),vec.end()); vector<int> tab(5001,0); tab[0] = 1; int wynik =0; int ile = 0; for(int k=0;k<n;k++) {// // cout << "LOW"<<endl; vector<para> add; int il = 0; for(int i=vec[k]-1;i<=5000;i++) {// cout << *x <<endl; if(tab[i]>0){ wynik+=tab[i]; wynik%=MOD; if(i + vec[k] <= 5000) add.push_back({i + vec[k],tab[i]}); else il+=tab[i]; il%=MOD;} } wynik+=ile; wynik%=MOD; ile*=2; ile%=MOD; ile+=il; ile%=MOD; for(auto i: add) { tab[i.a] +=i.b; tab[i.a]%=MOD; } //cout << "WTF"<<endl; } //for(auto t:secik) // cout << t.a << " " << t.b<<endl; cout << wynik<<endl; } |