#include <bits/stdc++.h> using namespace std; #define mod 1000000007 long long zbio[2500010], pos[5010], r; pair <int, int> sts[1000100]; int ile_min(int poc, int kon) //licymy od poc include do kon excepy { int res=0; for (int k=poc;k<kon;k++) { res += zbio[k]; res %= mod; } return res; } void dod_zbio(int a) { int kon = 5000-a+2; r=0; for (int i=a-1; i<=kon ;i++) { if ( zbio[i] ) { sts[r].first=i+a; sts[r].second=zbio[i]; r++; } } for (int i=0;i<r;i++) { zbio[ sts[i].first ] += sts[i].second; zbio[ sts[i].first ] %= mod; } } int main() { ios_base::sync_with_stdio(0); int n; cin>>n; for (int i=0;i<n;i++) cin>>pos[i]; sort(pos, pos+n); if ( pos[0] == 1 ) zbio[1]++; else { cout<<"0"<<endl; return 0; } int sum=1, miniente=0, previous=1; for (int i=1;i<n;i++) { int akt = pos[i]; miniente += ile_min( previous, akt ); miniente%=mod; int ile_n_prz = ((( sum-miniente+mod)%mod) +zbio[akt-1])%mod; sum += ile_n_prz; sum %= mod; dod_zbio(akt); if ( akt==1 ) { sum+=1; zbio[1]++; } previous = akt; } cout<<sum %mod; 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 | #include <bits/stdc++.h> using namespace std; #define mod 1000000007 long long zbio[2500010], pos[5010], r; pair <int, int> sts[1000100]; int ile_min(int poc, int kon) //licymy od poc include do kon excepy { int res=0; for (int k=poc;k<kon;k++) { res += zbio[k]; res %= mod; } return res; } void dod_zbio(int a) { int kon = 5000-a+2; r=0; for (int i=a-1; i<=kon ;i++) { if ( zbio[i] ) { sts[r].first=i+a; sts[r].second=zbio[i]; r++; } } for (int i=0;i<r;i++) { zbio[ sts[i].first ] += sts[i].second; zbio[ sts[i].first ] %= mod; } } int main() { ios_base::sync_with_stdio(0); int n; cin>>n; for (int i=0;i<n;i++) cin>>pos[i]; sort(pos, pos+n); if ( pos[0] == 1 ) zbio[1]++; else { cout<<"0"<<endl; return 0; } int sum=1, miniente=0, previous=1; for (int i=1;i<n;i++) { int akt = pos[i]; miniente += ile_min( previous, akt ); miniente%=mod; int ile_n_prz = ((( sum-miniente+mod)%mod) +zbio[akt-1])%mod; sum += ile_n_prz; sum %= mod; dod_zbio(akt); if ( akt==1 ) { sum+=1; zbio[1]++; } previous = akt; } cout<<sum %mod; return 0; } |