#include<bits/stdc++.h> using namespace std; int n,a,pref[300005],dpa[2000004],dnp[2000004],dp[300005],lp = 0,Base = 1; set <int> S; map <int,int> Map; int dodajMOD(int A,int B) { return (A + B) % 1000000007; } void INSERT(int * tab,int x,int w) { while(x > 0) { tab[x] = dodajMOD(tab[x],w); x >>= 1; } return; } int ileMniejszychRownych(int * tab,int x) { int wyn = tab[x]; while(x > 0) { if((x & 1) == 1) { wyn = dodajMOD(wyn,tab[x - 1]); } x >>= 1; } return wyn; } int ileWiekszych(int * tab,int x) { int wyn = 0; while(x > 0) { if((x & 1) == 0) { wyn = dodajMOD(wyn,tab[x + 1]); } x >>= 1; } return wyn; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin>>n; for(int i = 1;i <= n; i++) { cin>>a; pref[i] = dodajMOD(pref[i - 1],a); } //skalowanie for(int i = 0;i <= n; i++) S.insert(pref[i]); for(set<int>::iterator it = S.begin();it != S.end(); it++) { int x = * it; Map[x] = lp; lp++; } // while(Base < lp) Base <<= 1; INSERT(dpa,Base,1); for(int i = 1;i <= n; i++) { int pos = Map[pref[i]] + Base; if((pref[i] & 1) == 0) { dp[i] = dodajMOD(ileMniejszychRownych(dpa,pos),ileWiekszych(dnp,pos)); INSERT(dpa,pos,dp[i]); } else { dp[i] = dodajMOD(ileMniejszychRownych(dnp,pos),ileWiekszych(dpa,pos)); INSERT(dnp,pos,dp[i]); } } cout<<dp[n]<<'\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 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | #include<bits/stdc++.h> using namespace std; int n,a,pref[300005],dpa[2000004],dnp[2000004],dp[300005],lp = 0,Base = 1; set <int> S; map <int,int> Map; int dodajMOD(int A,int B) { return (A + B) % 1000000007; } void INSERT(int * tab,int x,int w) { while(x > 0) { tab[x] = dodajMOD(tab[x],w); x >>= 1; } return; } int ileMniejszychRownych(int * tab,int x) { int wyn = tab[x]; while(x > 0) { if((x & 1) == 1) { wyn = dodajMOD(wyn,tab[x - 1]); } x >>= 1; } return wyn; } int ileWiekszych(int * tab,int x) { int wyn = 0; while(x > 0) { if((x & 1) == 0) { wyn = dodajMOD(wyn,tab[x + 1]); } x >>= 1; } return wyn; } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cin>>n; for(int i = 1;i <= n; i++) { cin>>a; pref[i] = dodajMOD(pref[i - 1],a); } //skalowanie for(int i = 0;i <= n; i++) S.insert(pref[i]); for(set<int>::iterator it = S.begin();it != S.end(); it++) { int x = * it; Map[x] = lp; lp++; } // while(Base < lp) Base <<= 1; INSERT(dpa,Base,1); for(int i = 1;i <= n; i++) { int pos = Map[pref[i]] + Base; if((pref[i] & 1) == 0) { dp[i] = dodajMOD(ileMniejszychRownych(dpa,pos),ileWiekszych(dnp,pos)); INSERT(dpa,pos,dp[i]); } else { dp[i] = dodajMOD(ileMniejszychRownych(dnp,pos),ileWiekszych(dpa,pos)); INSERT(dnp,pos,dp[i]); } } cout<<dp[n]<<'\n'; return 0; } |