2527: [Poi2011]Meteors
好坑的一题,二分答案,check成功的国家放左边,否则放右边,然后继续分治下去,树状数组维护就可以了。
然后就炸longlong了,中间记得check成功就退出。
#include<cstdio>
#include<vector>
using namespace std;
const int MAXN=3e5+5;
int n,m,q,Now,c[MAXN],que[MAXN],q1[MAXN],q2[MAXN],Ans[MAXN];
long long F[MAXN];
struct Query{int L,R,p;}Q[MAXN];
vector<int> a[MAXN];
#include<cctype>
int read(){
int ret=0;char ch=getchar();bool f=1;
for(;!isdigit(ch);ch=getchar()) f^=!(ch^'-');
for(; isdigit(ch);ch=getchar()) ret=(ret<<1)+(ret<<3)+ch-48;
return f?ret:-ret;
}
void Put(int x,int p){for(;x<=m;x+=x&-x) F[x]+=p;}
long long Get(int x){long long Sum=0;for(;x;x-=x&-x) Sum+=F[x];return Sum;}
void Updata(int x,int f){
if(Q[x].L<=Q[x].R) Put(Q[x].L,f*Q[x].p),Put(Q[x].R+1,-f*Q[x].p);
else Put(1,f*Q[x].p),Put(Q[x].R+1,-f*Q[x].p),Put(Q[x].L,f*Q[x].p);
}
void Solve(int s,int t,int L,int R){
if(s>t) return;
if(L==R){for(int i=s;i<=t;i++) Ans[que[i]]=L;return;}
int mid=(R+L)>>1;
while(Now<mid) Updata(++Now,1);
while(Now>mid) Updata(Now--,-1);
int Top_q1=0,Top_q2=0;
for(int i=s;i<=t;i++){
long long Sum=0;
for(int j=0;j<a[que[i]].size()&&Sum<c[que[i]];j++) Sum+=Get(a[que[i]][j]);
if(Sum>=c[que[i]]) q1[++Top_q1]=que[i];else q2[++Top_q2]=que[i];
}
for(int i=1;i<=Top_q1;i++) que[s+i-1]=q1[i];
for(int i=1;i<=Top_q2;i++) que[s+Top_q1+i-1]=q2[i];
Solve(s,s+Top_q1-1,L,mid);Solve(s+Top_q1,t,mid+1,R);
}
int main(){
n=read(),m=read();Now=0;
for(int i=1;i<=m;i++) a[read()].push_back(i);
for(int i=1;i<=n;i++) c[i]=read(),que[i]=i;
q=read();
for(int i=1;i<=q;i++) Q[i]=(Query){read(),read(),read()};
Solve(1,n,1,q+1);
for(int i=1;i<=n;i++) if(Ans[i]==q+1) printf("NIE\n");else printf("%d\n",Ans[i]);
return 0;
}