#include <bits/stdc++.h> using namespace std; unsigned int t[5002]; unsigned int k[5002]; unsigned int mod=1000000007; int main() { ios_base::sync_with_stdio(0); cin.tie(0); long long sum; int n; sum=0; cin>>n; for(int i=0;i<n;++i) { cin>>t[i]; } sort(t,t+n); for(int i=0;i<n;++i) { if(t[i]==5000) { k[5001]+=k[5001]; k[5001]%=mod; k[5000]+=k[5000]; k[5000]%=mod; k[4999]+=k[4999]; k[4999]%=mod; } else { for(int j=5001;j+1>=t[i];--j) { if(t[i]+j>5000) { k[5001]+=k[j]; k[5001]%=mod; } else { k[t[i]+j]+=k[j]; k[t[i]+j]%=mod; } } if(t[i]==1) { k[1]++; } } } for(int i=0;i<=5001;++i) { sum+=k[i]; sum%=mod; } cout<<sum<<'\n'; return 0; }
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 | #include <bits/stdc++.h> using namespace std; unsigned int t[5002]; unsigned int k[5002]; unsigned int mod=1000000007; int main() { ios_base::sync_with_stdio(0); cin.tie(0); long long sum; int n; sum=0; cin>>n; for(int i=0;i<n;++i) { cin>>t[i]; } sort(t,t+n); for(int i=0;i<n;++i) { if(t[i]==5000) { k[5001]+=k[5001]; k[5001]%=mod; k[5000]+=k[5000]; k[5000]%=mod; k[4999]+=k[4999]; k[4999]%=mod; } else { for(int j=5001;j+1>=t[i];--j) { if(t[i]+j>5000) { k[5001]+=k[j]; k[5001]%=mod; } else { k[t[i]+j]+=k[j]; k[t[i]+j]%=mod; } } if(t[i]==1) { k[1]++; } } } for(int i=0;i<=5001;++i) { sum+=k[i]; sum%=mod; } cout<<sum<<'\n'; return 0; } |