zoukankan      html  css  js  c++  java
  • BZOJ2527 [Poi2011]Meteors 【整体二分 + 树状数组】

    题目

    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。

    输入样例

    3 5

    1 3 2 1 3

    10 5 7

    3

    4 2 4

    1 3 1

    3 5 2

    输出样例

    3

    NIE

    1

    提示

    1<=n,m,k<=3*10^5
    1<=Pi<=10^9
    1<=Ai<10^9

    题解

    显然我们需要二分每个国家最早到达需求的时刻,但我们不可能每次都从头统计一遍

    那就整体二分就搞定了√

    具体地,每次对一个区间前半的操作进行操作,然后查看当前区间的询问是否满足,不满足就丢到左区间继续查询,否则就减去当前值后丢到右区间继续查询

    #include<iostream>
    #include<cstdio>
    #include<cmath>
    #include<vector>
    #include<cstring>
    #include<algorithm>
    #define lbt(x) (x & -x)
    #define LL long long int
    #define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
    #define REP(i,n) for (int i = 1; i <= (n); i++)
    #define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");
    using namespace std;
    const int maxn = 300005,maxm = 10000005,INF = 1000000000;
    inline int read(){
    	int out = 0,flag = 1; char c = getchar();
    	while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
    	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
    	return out * flag;
    }
    int n,m,K,q[maxn],p[maxn],t[maxn],ans[maxn];
    LL S[maxn];
    vector<int> at[maxn];
    struct node{int l,r,v;}e[maxn];
    void add(int u,LL v){while (u <= m) S[u] += v,u += lbt(u);}
    LL query(int u){LL ans = 0; while (u) ans += S[u],u -= lbt(u); return ans;}
    void solve(int l,int r,int L,int R){
    	if (L > R) return;
    	if (l == r){
    		for (int i = L; i <= R; i++) ans[q[i]] = l;
    		return;
    	}
    	int mid = l + r >> 1;
    	for (int i = l; i <= mid && i <= K; i++){
    		if (e[i].l <= e[i].r){
    			add(e[i].l,e[i].v);
    			add(e[i].r + 1,-e[i].v);
    		}
    		else {
    			add(1,e[i].v);
    			add(e[i].r + 1,-e[i].v);
    			add(e[i].l,e[i].v);
    		}
    	}
    	int li = L,ri = R;
    	LL tmp;
    	for (int i = L; i <= R; i++){
    		int u = q[i];
    		tmp = 0;
    		for (unsigned int j = 0; j < at[u].size(); j++)
    			tmp += query(at[u][j]);
    		if (tmp < p[u]) p[u] -= tmp,t[ri--] = u;
    		else t[li++] = u;
    	}
    	for (int i = L; i <= R; i++) q[i] = t[i];
    	for (int i = l; i <= mid && i <= K; i++){
    		if (e[i].l <= e[i].r){
    			add(e[i].l,-e[i].v);
    			add(e[i].r + 1,e[i].v);
    		}
    		else {
    			add(1,-e[i].v);
    			add(e[i].r + 1,e[i].v);
    			add(e[i].l,-e[i].v);
    		}
    	}
    	solve(l,mid,L,li - 1); solve(mid + 1,r,ri + 1,R);
    }
    int main(){
    	n = read(); m = read();
    	for (int i = 1; i <= m; i++){
    		int u = read();
    		at[u].push_back(i);
    	}
    	for (int i = 1; i <= n; i++) p[i] = read();
    	K = read();
    	for (int i = 1; i <= K; i++){
    		e[i].l = read();
    		e[i].r = read();
    		e[i].v = read();
    	}
    	for (int i = 1; i <= n; i++) q[i] = i;
    	solve(1,K + 1,1,n);
    	for (int i = 1; i <= n; i++){
    		if (ans[i] == K + 1) puts("NIE");
    		else printf("%d
    ",ans[i]);
    	}
    	return 0;
    }
    
    
  • 相关阅读:
    POJ 3468 A Simple Problem with Integers
    BZOJ 4430 Guessing Camels
    POJ 2309 BST
    POJ 1990 MooFest
    cf 822B Crossword solving
    cf B. Black Square
    cf 828 A. Restaurant Tables
    Codefroces 822C Hacker, pack your bags!
    [HDU 2255] 奔小康赚大钱
    [BZOJ 1735] Muddy Fields
  • 原文地址:https://www.cnblogs.com/Mychael/p/8762410.html
Copyright © 2011-2022 走看看