洛谷时限比较小,目前超时一个点; 正在联系kkk; 我们要求对于每一个国家,它最早完成时间; 那么我们二分完成时间,并且把答案在这个二分区间内的国家也一起进入二分; 能不用线段树就不用; 区间加法可以用前缀和实现的;
#include#include #include #include #define Ll long longusing 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 >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]);}