传送门(内校oj)
题面:
SOL
整体二分即可,二分第几波陨石
(各种写挂。。。以后不复制粘贴了)
CODE
代码有点丑。。
#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define pf printf
#define ll long long
const int N=3e5+10;
struct node{
int l,r,k,id,op;
//1:query 2:update : tim:country
}q[N*3],q1[N*3],q2[N*3];
int ans[N],n,m,K,cnt;
ll req[N];
vector<int> G[N];
struct tree{
#define lb(k) (k&(-k))
ll sum[N];
inline void init(){memset(sum,0,sizeof sum);}
inline void update(int p,int k){
for(;p<=m;p+=lb(p))sum[p]+=k;
}
inline ll query(int p){
ll res=0;
for(;p>=1;p-=lb(p))res+=sum[p];
return res;
}
}tr;
void solve(int l,int r,int st,int ed){
if(l==r){
for(int i=st;i<=ed;++i)if(q[i].op)ans[q[i].id]=l;
return;
}
int mid=(l+r)>>1,cnt1=0,cnt2=0;
for(int i=st;i<=ed;++i){
if(q[i].op){
// cout<<"id:"<<q[i].id<<'\n';
ll res=0;
for(int j=0;j<G[q[i].id].size();++j){
res+=tr.query(G[q[i].id][j]);
if(res>=req[q[i].id])break;
}
// cout<<req[q[i].id]<<"->"<<res<<'\n';
if(res>=req[q[i].id])q1[++cnt1]=q[i];
else req[q[i].id]-=res,q2[++cnt2]=q[i];
}
else{
if(q[i].id<=mid)tr.update(q[i].l,q[i].k),tr.update(q[i].r+1,-q[i].k),q1[++cnt1]=q[i];
else q2[++cnt2]=q[i];
}
}
for(int i=1;i<=cnt1;++i)if(!q1[i].op)tr.update(q1[i].l,-q1[i].k),tr.update(q1[i].r+1,q1[i].k);
for(int i=1;i<=cnt1;++i)q[st+i-1]=q1[i];
for(int i=1;i<=cnt2;++i)q[st+cnt1+i-1]=q2[i];
solve(l,mid,st,st+cnt1-1);solve(mid+1,r,st+cnt1,ed);
}
signed main(){
sf("%d%d",&n,&m);
for(int i=1;i<=m;++i){
int k;sf("%d",&k);
G[k].push_back(i);
}
for(int i=1;i<=n;++i){
sf("%lld",&req[i]);
}
sf("%d",&K);
for(int i=1;i<=K;++i){
int l,r,k;sf("%d%d%d",&l,&r,&k);
if(l<=r){
q[++cnt]=(node){l,r,k,i,0};
tr.update(l,k);
tr.update(r+1,-k);
}
else{
swap(l,r);
q[++cnt]=(node){r,m,k,i,0};
q[++cnt]=(node){1,l,k,i,0};
tr.update(r,k);
tr.update(1,k);
tr.update(l+1,-k);
}
}
for(int i=1;i<=n;++i){
ll res=0;
for(int j=0;j<G[i].size();++j){
res+=tr.query(G[i][j]);
if(res>=req[i])break;
}
if(res<req[i])ans[i]=-1;
else q[++cnt]=(node){0,0,0,i,1};
}
tr.init();
solve(1,K,1,cnt);
for(int i=1;i<=n;++i){
if(ans[i]^(-1))cout<<ans[i]<<'\n';
else cout<<"NIE\n";
}
return 0;
}