#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#ifdef LOCAL
#define debug(...) __VA_ARGS__
#else
#define debug(...) {}
#endif
const ll mod = 1e9+7;
const int base = (1<<19);
ll tree[2*base][2];
pair<ll,int> pref[300001];
ll tab[300001];
pair<int,bool> where[300001];
void change(int v, ll x, bool y){
v += base;
tree[v][y] = (tree[v][y]+x)%mod;
v /= 2;
while (v){
tree[v][0] = (tree[2*v][0]+tree[2*v+1][0])%mod;
tree[v][1] = (tree[2*v][1]+tree[2*v+1][1])%mod;
v /= 2;
}
}
int query(int a, int b, bool y){
a += base-1;
b += base+1;
ll suma = 0;
while (b-a > 1){
if (!(a%2)) suma = (suma+tree[a+1][y])%mod;
if (b%2) suma = (suma+tree[b-1][y])%mod;
a /= 2;
b /= 2;
}
return suma;
}
int main(){
ios_base::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int i;
int n;
cin>>n;
change(0,1,0);
pref[0] = {0,0};
for (i = 1; i < n+1; i++){
cin>>tab[i];
pref[i].first = (pref[i-1].first+tab[i])%mod;
pref[i].second = i;
}
stable_sort(pref,pref+n+1);
int pointer = 0;
for (i = 0; i < n+1; i++){
if (i && pref[i].first != pref[i-1].first) pointer++;
where[pref[i].second] = {pointer,pref[i].first%2};
//cout<<pref[i].second<<" # "<<pref[i].first<<" # "<<where[pref[i].second].first<<" "<<where[pref[i].second].second<<"\n";
}
ll aktwy = 0;
for (i = 1; i < n+1; i++){
aktwy = query(0,where[i].first,where[i].second)%mod;
aktwy = (aktwy+query(where[i].first+1,n+1,where[i].second^1))%mod;
change(where[i].first,aktwy,where[i].second);
}
cout<<aktwy<<"\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 | #include <bits/stdc++.h> using namespace std; typedef long long ll; #ifdef LOCAL #define debug(...) __VA_ARGS__ #else #define debug(...) {} #endif const ll mod = 1e9+7; const int base = (1<<19); ll tree[2*base][2]; pair<ll,int> pref[300001]; ll tab[300001]; pair<int,bool> where[300001]; void change(int v, ll x, bool y){ v += base; tree[v][y] = (tree[v][y]+x)%mod; v /= 2; while (v){ tree[v][0] = (tree[2*v][0]+tree[2*v+1][0])%mod; tree[v][1] = (tree[2*v][1]+tree[2*v+1][1])%mod; v /= 2; } } int query(int a, int b, bool y){ a += base-1; b += base+1; ll suma = 0; while (b-a > 1){ if (!(a%2)) suma = (suma+tree[a+1][y])%mod; if (b%2) suma = (suma+tree[b-1][y])%mod; a /= 2; b /= 2; } return suma; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int i; int n; cin>>n; change(0,1,0); pref[0] = {0,0}; for (i = 1; i < n+1; i++){ cin>>tab[i]; pref[i].first = (pref[i-1].first+tab[i])%mod; pref[i].second = i; } stable_sort(pref,pref+n+1); int pointer = 0; for (i = 0; i < n+1; i++){ if (i && pref[i].first != pref[i-1].first) pointer++; where[pref[i].second] = {pointer,pref[i].first%2}; //cout<<pref[i].second<<" # "<<pref[i].first<<" # "<<where[pref[i].second].first<<" "<<where[pref[i].second].second<<"\n"; } ll aktwy = 0; for (i = 1; i < n+1; i++){ aktwy = query(0,where[i].first,where[i].second)%mod; aktwy = (aktwy+query(where[i].first+1,n+1,where[i].second^1))%mod; change(where[i].first,aktwy,where[i].second); } cout<<aktwy<<"\n"; return 0; } |
English