整體二分——[Poi2011]Meteors
阿新 • • 發佈:2019-02-08
https://www.luogu.org/problem/show?pid=3527
洛谷時限比較小,目前超時一個點;
正在聯絡kkk;
我們要求對於每一個國家,它最早完成時間;
那麼我們二分完成時間,並且把答案在這個二分割槽間內的國家也一起進入二分;
能不用線段樹就不用;
區間加法可以用字首和實現的;
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define Ll long long
using namespace std;
const int N=300005;
struct cs{int to,nxt;}a[N];
struct cz{int l,r,v;}c[N];
int head[N],ll;
int p[N],o[N],oo[N],ans[N];
bool ok[N];
Ll f[N];
int n,m,x,t,nn;
void init(int i,int v){for(;i<=m;i+=i&-i)f[i]+=v;}
Ll outit(int i){Ll ans=0;for(;i;i-=i&-i)ans+=f[i];return ans;}
void add(int k,int kk){
int l=c[k].l,r=c[k].r,v=c[k].v*kk;
if(r>=l)init(l,v),init(r+1,-v);else
init(1,v),init(r+1,-v),init(l,v);
}
void er(int L,int R,int l,int r){
if(r<l)return;
if(L==R){for(int i=l;i<=r;i++)ans[o[i]]=L;return;}
int mid=L+R>>1;
while(t<mid)add(++t, 1);//一種新的方式,感覺很牛逼
while (t>mid)add(t--,-1);
for(int i=l;i<=r;i++){
int x=o[i]; oo[i]=x; Ll sum=0;
for(int k=head[x];k;k=a[k].nxt){
sum+=outit(a[k].to);
if(sum>=p[x])break;
}
if(sum>=p[x])ok[x]=1;else ok[x]=0;
}
int top=l-1,sum=0;
for(int i=l;i<=r;i++)if(ok[oo[i]])o[++top]=oo[i];
for(int i=l;i<=r;i++)if(!ok[oo[i]])o[++sum+top]=oo[i];
er(L,mid ,l,top );
er(mid+1,R,top+1,r);
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
scanf("%d",&x);
a[++ll].to=i;
a[ll].nxt=head[x];
head[x]=ll;
}
for(int i=1;i<=n;i++)scanf("%d",&p[i]);
scanf("%d",&nn);
for(int i=1;i<=nn;i++)scanf("%d%d%d",&c[i].l,&c[i].r,&c[i].v);
c[++nn].l=1;c[nn].r=m;c[nn].v=1e9;
for(int i=1;i<=n;i++)o[i]=i;
er(1,nn,1,n);
for(int i=1;i<=n;i++)if(ans[i]==nn)printf("NIE\n");else printf("%d\n",ans[i]);
}