BZOJ 2527 & 整体二分

Byteotian Interstellar Union有N个成员国。现在它发现了一颗新的星球，这颗星球的轨道被分为M份（第M份和第1份相邻），第i份上有第Ai个国家的太空站。

这个星球经常会下陨石雨。BIU已经预测了接下来K场陨石雨的情况。
BIU的第i个成员国希望能够收集Pi单位的陨石样本。你的任务是判断对于每个国家，它需要在第几次陨石雨之后，才能收集足够的陨石。
输入：
第一行是两个数N,M。
第二行有M个数，第i个数Oi表示第i段轨道上有第Oi个国家的太空站。
第三行有N个数，第i个数Pi表示第i个国家希望收集的陨石数量。
第四行有一个数K，表示BIU预测了接下来的K场陨石雨。
接下来K行，每行有三个数Li,Ri,Ai，表示第K场陨石雨的发生地点在从Li顺时针到Ri的区间中（如果Li<=Ri，就是Li,Li+1,...,Ri，否则就是Ri,Ri+1,...,m-1,m,1,...,Li），向区间中的每个太空站提供Ai单位的陨石样本。
输出：
N行。第i行的数Wi表示第i个国家在第Wi波陨石雨之后能够收集到足够的陨石样本。如果到第K波结束后仍然收集不到，输出NIE。
数据范围：

　　数据范围： 1<=n,m,k<=3*10^5 1<=Pi<=10^9 1<=Ai<10^9

SOL:

　　这貌似比区间第k小还简单一点...流星雨已经按照时间顺序排好,也没有那么多繁琐的东西,而我们只要二分每个国家答案即可.非常方便统计.

　　然而我居然傻逼地觉得每次统计的时候暴力统计能过...给60s卡了评测好几次...

　　我会说我至今不知道怎么套上树状数组么...

CODE:

　　先贴一个T了的code,对拍貌似应该没有什么问题-----60s实现这是逗比吗?

　　

/*==========================================================================
# Last modified: 2016-02-26 13:59
# Filename: 2527.cpp
# Description:
==========================================================================*/
#define me AcrossTheSky
#include <cstdio>
#include <cmath>
#include <ctime>
#include <string>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm> 

#include <set>
#include <map>
#include <stack>
#include <queue>
#include <vector> 

#define lowbit(x) (x)&(-x)
#define FOR(i,a,b) for((i)=(a);(i)<=(b);(i)++)
#define FORP(i,a,b) for(int i=(a);i<=(b);i++)
#define FORM(i,a,b) for(int i=(a);i>=(b);i--)
#define ls(a,b) (((a)+(b)) << 1)
#define rs(a,b) (((a)+(b)) >> 1)
#define getlc(a) ch[(a)][0]
#define getrc(a) ch[(a)][1] 

#define maxn 400000
#define maxm 400000
#define pi 3.1415926535898
#define _e 2.718281828459
#define INF 1070000000
using namespace std;
typedef long long ll;
typedef unsigned long long ull; 

template<class T> inline
void read(T& num) {
    bool start=false,neg=false;
    char c;
    num=0;
    while((c=getchar())!=EOF) {
        if(c=='-') start=neg=true;
        else if(c>='0' && c<='9') {
            start=true;
            num=num*10+c-'0';
        } else if(start) break;
    }
    if(neg) num=-num;
}
/*==================split line==================*/
int n,m;
struct fall{
	int l,r,meter,id;
}q[maxm];
struct Infor{
	int k,pos;
}a[maxn],t[maxn];
int ans[maxn],tmp[maxn],o[maxm],cap[maxn];
int num=0;
bool mark[maxn];

void solve(int l,int r,int L,int R){
	/*printf("l:%d r:%d,L:%d,R:%d\n",l,r,L,R);
	FORP(i,l,r) printf("%d ",a[i].pos);
	cout << endl;*/
	if (l>r) return;
	if (L==R) {
		FORP(i,l,r)
			ans[a[i].pos]=L;
		return;
	}
	int mid=(L+R)>>1;
	memset(tmp,0,sizeof(tmp));
	for (int i=L;i<=mid;i++)
		if (q[i].r<q[i].l){
			FORP(j,q[i].l,m) if (tmp[o[j]]<cap[o[j]])
								tmp[o[j]]+=q[i].meter;
			FORP(j,1,q[i].r) if (tmp[o[j]]<cap[o[j]])
								tmp[o[j]]+=q[i].meter;
		}
		else FORP(j,q[i].l,q[i].r)
			if (tmp[o[j]]<cap[o[j]]) tmp[o[j]]+=q[i].meter;

	int cnt=0;
	FORP(i,l,r)
		if (a[i].k>tmp[a[i].pos])
		mark[i]=true,a[i].k-=tmp[a[i].pos],cap[a[i].pos]-=tmp[a[i].pos];
			else mark[i]=false,cnt++;
	int la=l,lb=l+cnt;
	FORP(i,l,r){
		if (mark[i]) t[lb++]=a[i];
		else t[la++]=a[i];
	}
	FORP(i,l,r) a[i]=t[i];
	solve(l,la-1,L,mid);
	solve(la,lb-1,mid+1,R);
}
int main(){
	freopen("a.in","r",stdin);
	freopen("tmp.out","w",stdout);
	 read(n); read(m);
	 FORP(i,1,m) read(o[i]);
	 FORP(i,1,n) {
	 	read(cap[i]);
	 	a[i].k=cap[i]; a[i].pos=i;
	}

	 int p; read(p);
	 FORP(i,1,p){
	 	int x,y,d;
	 	read(x); read(y); read(d);
		q[++num].l=x,q[num].r=y,q[num].meter=d,q[num].id=i;
	 }
	 q[++num].l=1; q[num].r=m; q[num].meter=INF; q[num].id=num;
	 solve(1,n,1,num);
	 FORP(i,1,n) {
	 	if (ans[i]==num) printf("NIE\n");
		else printf("%d\n",ans[i]);
	}
}