#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; } |
English