#pragma GCC optimize ("O3")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int, int> ii;
typedef vector<int> vi;
typedef double K;
constexpr int INF = 0x3f3f3f3f;
#define FOR(i, b, e) for(int i = (b); i < (e); i++)
#define TRAV(x, a) for(auto &x: (a))
#define SZ(x) ((int)(x).size())
#define PB push_back
#define X first
#define Y second
constexpr int mod = 1e9 + 7;
constexpr int T = 1 << 20;
ii tree[T * 2 + 5];
int shf;
vi ids;
ii ask(int l, int r) {
l = lower_bound(ids.begin(), ids.end(), l) - ids.begin();
r = lower_bound(ids.begin(), ids.end(), r) - ids.begin();
ll a = 0, b = 0;
for(l += T, r += T; l <= r; l >>= 1, r >>= 1) {
if(l & 1) a += tree[l].X, b += tree[l].Y, l++;
if((r & 1) ^ 1) a += tree[r].X, b += tree[r].Y, r--;
}
return {a % mod, b % mod};
}
void add(int v, int val) {
bool flag = v & 1;
v = lower_bound(ids.begin(), ids.end(), v) - ids.begin();
if(flag) for(v += T; v; v >>= 1) tree[v].Y = (tree[v].Y + val) % mod;
else for(v += T; v; v >>= 1) tree[v].X = (tree[v].X + val) % mod;
}
int ile() {
ii ask1 = ask(0, shf - 1);
ii ask2 = ask(shf, mod - 1);
if(shf & 1) return (ask1.X + ask2.Y) % mod;
return (ask1.Y + ask2.X) % mod;
}
void solve() {
int n;
cin >> n;
vi vec(n);
TRAV(x, vec) cin >> x;
ids = {0, mod - 1};
TRAV(x, vec) shf = (shf + mod - x) % mod, ids.PB(shf), ids.PB(shf - 1);
sort(ids.begin(), ids.end());
ids.erase(unique(ids.begin(), ids.end()), ids.end());
shf = 0;
add(0, 1);
FOR(i, 0, n) {
shf = (shf + mod - vec[i]) % mod;
if(i + 1 == n) cout << ile() << '\n';
else add(shf, ile());
}
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
// int tt; cin >> tt;
// FOR(te, 0, tt) {
// // cout << "Case #" << te + 1 << ": ";
// solve();
// }
solve();
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 | #pragma GCC optimize ("O3") #include <bits/stdc++.h> using namespace std; typedef long long ll; typedef pair<int, int> ii; typedef vector<int> vi; typedef double K; constexpr int INF = 0x3f3f3f3f; #define FOR(i, b, e) for(int i = (b); i < (e); i++) #define TRAV(x, a) for(auto &x: (a)) #define SZ(x) ((int)(x).size()) #define PB push_back #define X first #define Y second constexpr int mod = 1e9 + 7; constexpr int T = 1 << 20; ii tree[T * 2 + 5]; int shf; vi ids; ii ask(int l, int r) { l = lower_bound(ids.begin(), ids.end(), l) - ids.begin(); r = lower_bound(ids.begin(), ids.end(), r) - ids.begin(); ll a = 0, b = 0; for(l += T, r += T; l <= r; l >>= 1, r >>= 1) { if(l & 1) a += tree[l].X, b += tree[l].Y, l++; if((r & 1) ^ 1) a += tree[r].X, b += tree[r].Y, r--; } return {a % mod, b % mod}; } void add(int v, int val) { bool flag = v & 1; v = lower_bound(ids.begin(), ids.end(), v) - ids.begin(); if(flag) for(v += T; v; v >>= 1) tree[v].Y = (tree[v].Y + val) % mod; else for(v += T; v; v >>= 1) tree[v].X = (tree[v].X + val) % mod; } int ile() { ii ask1 = ask(0, shf - 1); ii ask2 = ask(shf, mod - 1); if(shf & 1) return (ask1.X + ask2.Y) % mod; return (ask1.Y + ask2.X) % mod; } void solve() { int n; cin >> n; vi vec(n); TRAV(x, vec) cin >> x; ids = {0, mod - 1}; TRAV(x, vec) shf = (shf + mod - x) % mod, ids.PB(shf), ids.PB(shf - 1); sort(ids.begin(), ids.end()); ids.erase(unique(ids.begin(), ids.end()), ids.end()); shf = 0; add(0, 1); FOR(i, 0, n) { shf = (shf + mod - vec[i]) % mod; if(i + 1 == n) cout << ile() << '\n'; else add(shf, ile()); } } int main() { ios::sync_with_stdio(0); cin.tie(0); // int tt; cin >> tt; // FOR(te, 0, tt) { // // cout << "Case #" << te + 1 << ": "; // solve(); // } solve(); return 0; } |
English