POI2011 Meteors(CDQ分治 + 整体二分)

/**
题意：有N个国家M个观察站，每个观察站只属于一个国家，观察站从1到M标号，M和1相邻，第i个国家需要Wi单位的陨石。
现在有k场陨石雨，第i场陨石雨波及的范围顺时针是Li~Ri，可以为波及的每个观察站提供Ai单位陨石。问每个国家，它需
要在第几次陨石雨之后，才能收集足够的陨石。如果K场陨石雨之后不能满足，则该国家输出NIE。

思路：CDQ分治 + 整体二分，大概思路就是对所有国家二分答案时间t， 对于时间在区间[l, r]的国家，时间[l, mid]的陨石
雨用树状数组维护， 在这段时间已经满足的国家放进[l, mid]中去， 不满足的放进[mid + 1, r]中去处理
*/

#include<bits/stdc++.h>
typedef long long ll;
const int maxn = 3e5 + 10;
const int INF = 1e9 + 10;
using namespace std;

int L[maxn], R[maxn], A[maxn], W[maxn];
int to[maxn], n, m, t, ans[maxn], s[maxn];
vector<int> G[maxn];
int can[maxn], notcan[maxn];
ll C[maxn];

void update(int x, int data) { for(; x <= m; x += x & -x) C[x] += data; }
ll get_sum(int x) { ll tot = 0; for(; x; x -= x & -x) tot += C[x]; return tot; }

void update(int l, int mid, int flag) {
    for(int i = l; i <= mid; i++) {
        int fr = L[i], ot = R[i];
        if(fr <= ot) { update(fr, flag * A[i]); update(ot + 1, -1 * flag * A[i]); }
        else {
            update(fr, flag * A[i]); update(m + 1, -1 * flag * A[i]);
            update(1, flag * A[i]); update(ot + 1, -1 * flag * A[i]);
        }
    }
}

void solve(int l, int r, int f, int t) {
    if(t == f) return ;
    int mid = (l + r) >> 1;
    update(l, mid, 1);
    int t1 = 0, t2 = 0;
    for(int i = f; i < t; i++) {
        int now_point = s[i];
        int flag = 0; ll tal = 0;
        for(int j = 0; j < G[now_point].size(); j++) {
            int idx = G[now_point][j];
            tal += get_sum(idx);
            if(tal >= W[now_point]) {  ans[now_point] = min(ans[now_point], mid); flag = 1; break; }
        }
        if(flag) can[t1++] = now_point;
        else notcan[t2++] = now_point;
    }
    int cnt = f;
    for(int i = 0; i < t1; i++) s[cnt++] = can[i];
    for(int i = 0; i < t2; i++) s[cnt++] = notcan[i];
    if(l == r) { update(l, mid, -1); return ; }
    solve(mid + 1, r, f + t1, t);
    update(l, mid, -1);
    solve(l, mid, f, f + t1);
}

int main() {
    while(scanf("%d %d", &n, &m) != EOF) {
        for(int i = 0; i < maxn; i++) G[i].clear();
        for(int i = 1; i <= m; i++) { scanf("%d", &to[i]); G[to[i]].push_back(i); }
        for(int i = 1; i <= n; i++) { scanf("%d", &W[i]); ans[i] = INF; s[i - 1] = i; }
        scanf("%d", &t);
        for(int i = 1; i <= t; i++) scanf("%d %d %d", &L[i], &R[i], &A[i]);
        memset(C, 0, sizeof C);
        solve(1, t, 0, n);
        for(int i = 1; i <= n; i++) {
            if(ans[i] != INF) printf("%d\n", ans[i]);
            else printf("NIE\n");
        }
    }
    return 0;
}