loj#6032. 「雅礼集训 2017 Day2」水箱(并查集 贪心 扫描线)

题意

sol

神仙题+%%%%%%%%

我再来复述一遍。。

首先按照\(y\)坐标排序，然后维护一个扫描线从低处往高处考虑。

一个连通块的内状态使用两个变量即可维护\(ans\)表示联通块内的最大答案，\(f\)表示联通块内\(k=1\)的数量

若当前的水超过了当前的挡板，那么将当前联通块和下一个位置所在的联通块合并。

若是一个\(k=0\)的操作，则一定满足。

若是\(k=1\)的操作，那么就将\(f++\)，然后更新一下答案。

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
const int maxn = 1e6 + 10, inf = 1e9 + 7, mod = 998244353;
template <typename a, typename b> inline bool chmin(a &a, b b){if(a > b) {a = b; return 1;} return 0;}
template <typename a, typename b> inline bool chmax(a &a, b b){if(a < b) {a = b; return 1;} return 0;}
template <typename a, typename b> inline ll add(a x, b y) {if(x + y < 0) return x + y + mod; return x + y >= mod ? x + y - mod : x + y;}
template <typename a, typename b> inline void add2(a &x, b y) {if(x + y < 0) x = x + y + mod; else x = (x + y >= mod ? x + y - mod : x + y);}
template <typename a, typename b> inline ll mul(a x, b y) {return 1ll * x * y % mod;}
template <typename a, typename b> inline void mul2(a &x, b y) {x = (1ll * x * y % mod + mod) % mod;}
inline int read() {
    char c = getchar(); int x = 0, f = 1;
    while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x * f;
}
int n, m, cnt, ans[maxn], f[maxn], fa[maxn];
int find(int x) {
    return fa[x] ? fa[x] = find(fa[x]) : x;
}
struct query {
    int opt, x, y;
    bool operator < (const query &rhs) const {
        return y == rhs.y ? opt < rhs.opt : y < rhs.y;
    }
}q[maxn];
void solve() {
    memset(ans, 0, sizeof(ans)); memset(f, 0, sizeof(f)); memset(fa, 0, sizeof(fa));
    cnt = 0;
    n = read(); m = read();
    for(int i = 1; i < n; i++) q[++cnt] = {-1, i, read()};
    for(int i = 1; i <= m; i++) q[++cnt].x = read(), q[cnt].y = read(), q[cnt].opt = read();
    stable_sort(q + 1, q + cnt + 1);
    int ret = 0;
    for(int i = 1; i <= cnt; i++) {
        int op = q[i].opt, x = q[i].x;
        if(op == -1) {
            int y = find(x + 1); x = find(x);
            fa[y] = x; f[x] += f[y]; ans[x] += ans[y]; chmax(ret, ans[x]);
        } else if(op == 0) {
            chmax(ret, ++ans[find(x)]);
        } else {
            x = find(x); chmax(ans[x], ++f[x]);
            chmax(ret, ans[x]);
        }
    }
    cout << ret << '\n';
}
int main() {
    for(int t = read(); t--; solve());
    return 0;
}