#include <bits/stdc++.h> using namespace std; #define rep(i,a,n) for (int i=a;i<n;i++) #define per(i,a,n) for (int i=n-1;i>=a;i--) #define pb push_back #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define SZ(x) ((int)(x).size()) typedef vector<int> VI; typedef long long ll; typedef pair<int,int> PII; const ll mod=1000000007; ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;} ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;} // head const int N=201000; ll k,c,t[N],s[N],boom[N],ret[N]; vector<PII> e; set<pair<ll,int> > ev; set<PII> seg; int n,m,d; void add(int l,int r) { k+=(ll)(r-l)*(r-l+1)/2; c+=t[l]*(r-l+1)-(s[r]-s[l-1]); } void del(int l,int r) { k-=(ll)(r-l)*(r-l+1)/2; c-=t[l]*(r-l+1)-(s[r]-s[l-1]); } void gao(int id) { auto it=seg.lower_bound(mp(id,id)),pit=it; ++pit; auto nit=pit; ++nit; del(it->fi,it->se); del(pit->fi,pit->se); add(it->fi,pit->se); int l=it->fi,r=pit->se,rr=nit->fi; int md=pit->fi; ev.erase(mp(boom[md],md)); seg.erase(it); seg.erase(pit); seg.insert(mp(l,r)); boom[l]=(t[rr]-t[l])/(rr-l); ev.insert(mp(boom[l],l)); } int main() { scanf("%d%d",&n,&m); t[1]=0; rep(i,2,n+2) scanf("%lld",t+i),s[i]=s[i-1]+t[i]; ++n; rep(i,0,m) { scanf("%d",&d); e.pb(mp(d,i)); } sort(all(e)); t[n+1]=1ll<<60; seg.insert(mp(n+1,n+1)); rep(i,1,n+1) { add(i,i); seg.insert(mp(i,i)); boom[i]=t[i+1]-t[i]; ev.insert(mp(boom[i],i)); } rep(i,0,m) { while (ev.begin()->fi<e[i].fi) { int id=ev.begin()->se; ev.erase(ev.begin()); gao(id); } ret[e[i].se]=k*e[i].fi+c; } rep(i,0,m) printf("%lld\n",ret[i]); }
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 | #include <bits/stdc++.h> using namespace std; #define rep(i,a,n) for (int i=a;i<n;i++) #define per(i,a,n) for (int i=n-1;i>=a;i--) #define pb push_back #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define SZ(x) ((int)(x).size()) typedef vector<int> VI; typedef long long ll; typedef pair<int,int> PII; const ll mod=1000000007; ll powmod(ll a,ll b) {ll res=1;a%=mod; assert(b>=0); for(;b;b>>=1){if(b&1)res=res*a%mod;a=a*a%mod;}return res;} ll gcd(ll a,ll b) { return b?gcd(b,a%b):a;} // head const int N=201000; ll k,c,t[N],s[N],boom[N],ret[N]; vector<PII> e; set<pair<ll,int> > ev; set<PII> seg; int n,m,d; void add(int l,int r) { k+=(ll)(r-l)*(r-l+1)/2; c+=t[l]*(r-l+1)-(s[r]-s[l-1]); } void del(int l,int r) { k-=(ll)(r-l)*(r-l+1)/2; c-=t[l]*(r-l+1)-(s[r]-s[l-1]); } void gao(int id) { auto it=seg.lower_bound(mp(id,id)),pit=it; ++pit; auto nit=pit; ++nit; del(it->fi,it->se); del(pit->fi,pit->se); add(it->fi,pit->se); int l=it->fi,r=pit->se,rr=nit->fi; int md=pit->fi; ev.erase(mp(boom[md],md)); seg.erase(it); seg.erase(pit); seg.insert(mp(l,r)); boom[l]=(t[rr]-t[l])/(rr-l); ev.insert(mp(boom[l],l)); } int main() { scanf("%d%d",&n,&m); t[1]=0; rep(i,2,n+2) scanf("%lld",t+i),s[i]=s[i-1]+t[i]; ++n; rep(i,0,m) { scanf("%d",&d); e.pb(mp(d,i)); } sort(all(e)); t[n+1]=1ll<<60; seg.insert(mp(n+1,n+1)); rep(i,1,n+1) { add(i,i); seg.insert(mp(i,i)); boom[i]=t[i+1]-t[i]; ev.insert(mp(boom[i],i)); } rep(i,0,m) { while (ev.begin()->fi<e[i].fi) { int id=ev.begin()->se; ev.erase(ev.begin()); gao(id); } ret[e[i].se]=k*e[i].fi+c; } rep(i,0,m) printf("%lld\n",ret[i]); } |