bzoj 2243 染色(树链剖分)

题外话


首先这是个挺裸的题,由于太久没写剖分导致调了好久,前天调了一下午,一直查不到错
昨晚在看春晚的时候突然灵机一动,发现合并的时候出了问题,开电脑把它A掉了= =
感觉自己也蛮拼的

Description


nm2
1:abc
2:ab()"112221"3:"11""222""1"

Solution


很显然这是要剖分的辣,然后用线段树维护信息。然后我们会发现合并的时候有点蛋疼
它要求颜色段数量,所以我们很自然的想到要维护左右端的颜色信息来进行合并
线段树合并时 sum[rt]=sum[ls]+sum[rs](rc[ls]==lc[rs]) 即可
但是我们正常在查询的时候是要在剖分后的树上”跳来跳去的”
所以每次查询的时候我们都需要维护当前左右端颜色信息,然后在”跳”的过程中合并
注意颜色是 [0,109]1
合并的时候想清楚,注意细节就好辣!可以仔细想想或者详情参见代码QvQ

Code

#include <bits/stdc++.h>//树链剖分
using namespace std;
#define ls (rt << 1)
#define rs (rt << 1 | 1)
const int N = 100005, M = N << 2;
int n, m, lcol, rcol, tot, cnt, w[N], q[N], top[N], sz[N], son[N], pre[N], id[N], dep[N], col[N], lc[M], rc[M], sum[M], mark[M], to[M], nxt[M], head[N];
bool vis[N];
inline int readInt() {
    char c;
    while (c = getchar(), c < '0' || c > '9');
    int t = c - '0';
    while (c = getchar(), c >= '0' && c <= '9') t = t * 10 + c - '0';
    return t;
}
void add(int u, int v) {
    to[tot] = v, nxt[tot] = head[u], head[u] = tot++;
    to[tot] = u, nxt[tot] = head[v], head[v] = tot++;
}
void pushdown(int rt) {
    if (~mark[rt]) {
        lc[ls] = lc[rs] = rc[ls] = rc[rs] = mark[ls] = mark[rs] = mark[rt];
        sum[ls] = sum[rs] = 1;
        mark[rt] = -1;
    }
}
void pushup(int rt) {
    lc[rt] = lc[ls], rc[rt] = rc[rs];
    sum[rt] = sum[ls] + sum[rs] - (rc[ls] == lc[rs]);
}
void build(int rt, int l, int r) {
    mark[rt] = -1;
    if (l == r) {
        lc[rt] = rc[rt] = w[l];
        sum[rt] = 1;
        return ;
    }
    int mid = l + r >> 1;
    build(ls, l, mid);
    build(rs, mid + 1, r);
    pushup(rt);
}
void change(int rt, int l, int r, int L, int R, int c) {
    if (L <= l && R >= r) {
        lc[rt] = rc[rt] = mark[rt] = c;
        sum[rt] = 1;
        return;
    }
    pushdown(rt);
    int mid = l + r >> 1;
    if (L <= mid)   change(ls, l, mid, L, R, c);
    if (R > mid)    change(rs, mid + 1, r, L, R, c);
    pushup(rt);
}
int ask(int rt, int l, int r, int L, int R) {
    if (L == l) lcol = lc[rt];
    if (R == r) rcol = rc[rt];
    if (L <= l && R >= r)   return sum[rt];
    pushdown(rt);
    int mid = l + r >> 1, t = 0, t1 = -1, t2 = -1;
    if (L <= mid)   t += ask(ls, l, mid, L, R), t1 = rc[ls];
    if (R > mid)    t += ask(rs, mid + 1, r, L, R), t2 = lc[rs];
    t -= (t1 == t2 && ~t1);
    return t;
}
void work(int a, int b, int c) {
    while (top[a] != top[b]) {
        if (dep[top[a]] < dep[top[b]])  swap(a, b);
        change(1, 1, n, id[top[a]], id[a], c);
        a = pre[top[a]];    
    }
    if (dep[a] < dep[b])    swap(a, b);
    change(1, 1, n, id[b], id[a], c);
}
int work2(int a, int b) {
    int t = 0, t1 = -1, t2 = -1;
    while (top[a] != top[b]) {
        if (dep[top[a]] < dep[top[b]])  swap(a, b), swap(t1, t2);
        t += ask(1, 1, n, id[top[a]], id[a]);
        t -= (t1 == rcol && ~t1);
        t1 = lcol;
        a = pre[top[a]];
    }
    if (dep[a] < dep[b])    swap(a, b), swap(t1, t2);
    t += ask(1, 1, n, id[b], id[a]);
    t -= ((t1 == rcol && ~t1) + (t2 == lcol && ~t2));
    return t;
}
void gao() {
    int r = 0;
    vis[dep[1] = q[0] = 1] = 1;
    for (int i = 0; i <= r; ++i)
        for (int j = head[q[i]]; ~j; j = nxt[j]) 
            if (!vis[to[j]]){
                vis[to[j]] = 1;
                dep[q[++r] = to[j]] = dep[q[i]] + 1;
                pre[q[r]] = q[i];
            }
    for (int i = r; i >= 0; --i) {
        sz[pre[q[i]]] += ++sz[q[i]];
        if (sz[son[pre[q[i]]]] < sz[q[i]])  son[pre[q[i]]] = q[i];
    }
    for (int i = 0; i <= r; ++i)
        if (!top[q[i]]) {
            for (int j = q[i]; j; j = son[j]) {
                top[j] = q[i];
                w[id[j] = ++cnt] = col[j];
            }
        }
    build(1, 1, n);
}
int main() {
    n = readInt(), m = readInt();
    memset(head, -1, sizeof(head));
    for (int i = 1; i <= n; ++i)    col[i] = readInt();
    for (int i = 1, x, y; i < n; ++i) {
        x = readInt(), y = readInt();
        add(x, y);
    }
    gao();
    while (m--) {
        char s[5];
        scanf("%s", s);
        int a, b, c;
        if (s[0] == 'C') {
            a = readInt(), b = readInt(), c = readInt();
            work(a, b, c);
        }
        else {
            a = readInt(), b = readInt();
            printf("%d\n", work2(a, b));
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值