#include<bits/stdc++.h> //#pragma GCC optimize("Ofast") //#pragma GCC optimize("trapv") #define st first #define nd second #define pb(x) push_back(x) #define pp(x) pop_back(x) #define mp(a, b) make_pair(a, b) #define all(x) (x).begin(), (x).end() #define rev(x) reverse(all(x)) #define sor(x) sort(all(x)) #define sz(x) (int)(x).size() #define rsz(x) resize(x) using namespace std; ///~~~~~~~~~~~~~~~~~~~~~~~~~~ void debug(){cerr<<"\n";} template <typename H, typename... T> void debug(H h, T... t) {cerr<<h; if (sizeof...(t)) cerr << ", "; debug(t...);} #define deb(x...) cerr<<#x<<" = ";debug(x); ///~~~~~~~~~~~~~~~~~~~~~~~~~ typedef long long ll; typedef long double ld; typedef pair<int, int> pii; typedef pair<ll, ll> pll; typedef vector<int> vi; typedef vector<pii > vii; typedef vector<ll> vl; typedef vector<pll> vll; typedef string str; #define BOOST ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL); mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count()); const int N=3e5+5, INF=1e9+5, mod=1e9+7; int tab[2][N+N]; void add(int t, int id, int val){ for(id+=N; id>0; id>>=1){ tab[t][id]+=val; tab[t][id]%=mod; } } int sum(int t, int l, int r){ ll ans=0; for(l+=N, r+=N; l<r; l>>=1, r>>=1){ if(l&1)ans+=tab[t][l++]; if(r&1)ans+=tab[t][--r]; } return ans%mod; } int main(){ BOOST; int n; cin>>n; vl pref; pref.pb(0); for(int i=0; i<n; i++){ int x; cin>>x; pref.pb((pref.back()+x)%(mod)); } vl pref2=pref; sor(pref2); pref2.resize(unique(all(pref2))-pref2.begin()); add(0, 0, 1); for(int i=1; i<=n; i++){ ll dp=0; int k=lower_bound(all(pref2), pref[i])-pref2.begin(); dp+=sum(pref[i]&1, 0, k+1); dp+=sum((pref[i]+1)&1, k+1, pref2.size()); dp%=mod; add(pref[i]&1, k, dp); if(i==n)cout<<dp<<"\n"; } }
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 | #include<bits/stdc++.h> //#pragma GCC optimize("Ofast") //#pragma GCC optimize("trapv") #define st first #define nd second #define pb(x) push_back(x) #define pp(x) pop_back(x) #define mp(a, b) make_pair(a, b) #define all(x) (x).begin(), (x).end() #define rev(x) reverse(all(x)) #define sor(x) sort(all(x)) #define sz(x) (int)(x).size() #define rsz(x) resize(x) using namespace std; ///~~~~~~~~~~~~~~~~~~~~~~~~~~ void debug(){cerr<<"\n";} template <typename H, typename... T> void debug(H h, T... t) {cerr<<h; if (sizeof...(t)) cerr << ", "; debug(t...);} #define deb(x...) cerr<<#x<<" = ";debug(x); ///~~~~~~~~~~~~~~~~~~~~~~~~~ typedef long long ll; typedef long double ld; typedef pair<int, int> pii; typedef pair<ll, ll> pll; typedef vector<int> vi; typedef vector<pii > vii; typedef vector<ll> vl; typedef vector<pll> vll; typedef string str; #define BOOST ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL); mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count()); const int N=3e5+5, INF=1e9+5, mod=1e9+7; int tab[2][N+N]; void add(int t, int id, int val){ for(id+=N; id>0; id>>=1){ tab[t][id]+=val; tab[t][id]%=mod; } } int sum(int t, int l, int r){ ll ans=0; for(l+=N, r+=N; l<r; l>>=1, r>>=1){ if(l&1)ans+=tab[t][l++]; if(r&1)ans+=tab[t][--r]; } return ans%mod; } int main(){ BOOST; int n; cin>>n; vl pref; pref.pb(0); for(int i=0; i<n; i++){ int x; cin>>x; pref.pb((pref.back()+x)%(mod)); } vl pref2=pref; sor(pref2); pref2.resize(unique(all(pref2))-pref2.begin()); add(0, 0, 1); for(int i=1; i<=n; i++){ ll dp=0; int k=lower_bound(all(pref2), pref[i])-pref2.begin(); dp+=sum(pref[i]&1, 0, k+1); dp+=sum((pref[i]+1)&1, k+1, pref2.size()); dp%=mod; add(pref[i]&1, k, dp); if(i==n)cout<<dp<<"\n"; } } |