loj10141. 「一本通 4.5 练习 3」染色

思路:

  利用树链剖分转化为线段树问题,考虑线段上经这两种操作后的sum求值方法,并着重考虑树链合并的情况即可。线段树为了打cov,要注意将所有的颜色统一加一。

#include<cstdio>
#include<iostream>
using namespace std;
const int maxn = 100010;
inline void qread(int &x){
    x = 0;
    register int ch = getchar();
    while(ch < ‘0‘ || ch > ‘9‘)    ch = getchar();
    while(ch >= ‘0‘ && ch <= ‘9‘)    x = 10 * x + ch - 48, ch = getchar();
}
int n, q;
int head[maxn];
int go[maxn << 1];
int nxt[maxn << 1];

int val[maxn];
int size[maxn];
int son[maxn];
int deep[maxn];
int f[maxn];
int top[maxn];
int seg[maxn];
int rev[maxn];
int end[maxn];

int sum[maxn << 2];
int cov[maxn << 2];
inline void init(){
    qread(n);
    qread(q);
    for(int i=1; i<=n; ++i){
        qread(val[i]);
        ++val[i];
    }
    for(int i=1; i<n; ++i){
        int x, y;
        qread(x), qread(y);
        go[i] = y;
        nxt[i] = head[x];
        head[x] = i;
        go[i + n] = x;
        nxt[i + n] = head[y];
        head[y] = i + n;
    }
    deep[1] = top[1] = seg[0] = seg[1] = rev[1] = 1;
}
void dfs1(int x){
    size[x] = 1;
    for(register int i = head[x]; i; i = nxt[i])
        if(!deep[go[i]]){
            f[go[i]] = x;
            deep[go[i]] = deep[x] + 1;
            dfs1(go[i]);
            size[x] += size[go[i]];
            if(size[go[i]] > size[son[x]])
                son[x] = go[i];
        }
}
void dfs2(int x){
    end[x] = x;
    if(son[x]){
        top[son[x]] = top[x];
        seg[son[x]] = ++seg[0];
        rev[seg[0]] = son[x];
        dfs2(son[x]);
        end[x] = end[son[x]];
    }
    for(register int i = head[x]; i; i = nxt[i])
        if(!top[go[i]]){
            top[go[i]] = go[i];
            seg[go[i]] = ++seg[0];
            rev[seg[0]] = go[i];
            dfs2(go[i]);
            end[x] = end[go[i]];
        }
}
void build(int l, int r, int x){
    if(l == r){
        sum[x] = 1;
        return ;
    }
    int mid = (l + r) >> 1;
    build(l, mid, x << 1);
    build(mid + 1, r, x << 1 | 1);
    sum[x] = sum[x << 1] + sum[x << 1 | 1] - (val[rev[mid]] == val[rev[mid + 1]]);
}
inline void pushdown(int x){
    cov[x << 1] = cov[x];
    cov[x << 1 | 1] = cov[x];
    sum[x << 1] = 1;
    sum[x << 1 | 1] = 1;
    cov[x] = 0;
}
int visit(int l, int r, int P, int x){
    if(l == r)
        return cov[x] ? cov[x] : val[rev[l]];
    if(cov[x])    pushdown(x);
    int mid = (l + r) >> 1;
    if(mid >= P)    return visit(l, mid, P, x << 1);
    else            return visit(mid + 1, r, P, x << 1 | 1);
}
void change(int l, int r, int L, int R, int v, int x){
    if(L <= l && R >= r){
        cov[x] = v;
        sum[x] = 1;
        return ;
    }
    if(cov[x])    pushdown(x);
    int mid = (l + r) >> 1;
    if(mid >= L)    change(l, mid, L, R, v, x << 1);
    if(mid < R)        change(mid + 1, r, L, R, v, x << 1 | 1);
    sum[x] = sum[x << 1] + sum[x << 1 | 1] - (visit(1, n, mid, 1) == visit(1, n, mid + 1, 1));
}
int ask(int l, int r, int L, int R, int x){
    if(L <= l && R >= r)
        return sum[x];
    if(cov[x])    pushdown(x);
    int mid = (l + r) >> 1, cnt = 0, ans = 0;
    if(mid >= L)    ++cnt, ans += ask(l, mid, L, R, x << 1);
    if(mid < R)        ++cnt, ans += ask(mid + 1, r, L, R, x << 1 | 1);
    if(cnt == 2)    ans -= (visit(1, n, mid, 1) == visit(1, n, mid + 1, 1));
    return ans;
}
int treeask(int x, int y){
    int fx = top[x], fy = top[y], ans = 0;
    while(fx != fy){
        if(deep[fx] < deep[fy])    swap(x, y), swap(fx, fy);
        ans += ask(1, n, seg[fx], seg[x], 1);
        if(f[fx] != 0 && visit(1, n, seg[fx], 1) == visit(1, n, seg[f[fx]], 1))    ans--;
        x = f[fx];
        fx = top[x];
    }
    if(deep[x] > deep[y])    swap(x, y);
    ans += ask(1, n, seg[x], seg[y], 1);
    return ans;
}
void treechange(int x, int y, int z){
    int fx = top[x], fy = top[y];
    while(fx != fy){
        if(deep[fx] < deep[fy])    swap(x, y), swap(fx, fy);
        change(1, n, seg[fx], seg[x], z, 1);
        x = f[fx];
        fx = top[x];
    }
    if(deep[x] > deep[y])    swap(x, y);
    change(1, n, seg[x], seg[y], z, 1);
}

int main(void)
{
    init();
    dfs1(1);
    dfs2(1);
    build(1, n, 1);
    while(q--){
        string op;
        cin >> op;
        if(op == "C") {
            int a, b, c;
            qread(a), qread(b), qread(c);
            ++c;
            treechange(a, b, c);
        }
        else{
            int a, b;
            qread(a), qread(b);
            printf("%d\n", treeask(a, b));
        }
    }
}

原文地址:https://www.cnblogs.com/junk-yao-blog/p/9485071.html

时间: 2024-09-29 07:06:56

loj10141. 「一本通 4.5 练习 3」染色的相关文章

10249「一本通 1.3 例 5」weight

#10249「一本通 1.3 例 5」weight 题目描述 原题来自:USACO 已知原数列a1,a2,...,an中的前1项,前2项,前3项,... ,前 n 项的和,以及后 1 项,后 2 项,后 3 项,...,后 n 项的和,但是==所有的数都被打乱了顺序==.此外,我们还知道数列中的数存在于集合 S 中.试求原数列.当存在多组可能的数列时,求字典序最小的数列. 输入格式 第 1 行,一个整数 n . 第 2 行, 2 × n 个整数,注意:数据已被打乱. 第 3 行,一个整数 m ,

「一本通 4.1 练习 2」简单题

题目描述 题目来源:CQOI 2006 有一个 n 个元素的数组,每个元素初始均为 0.有 m 条指令,要么让其中一段连续序列数字反转--0 变 1,1变 0(操作 1),要么询问某个元素的值(操作 2). 例如当 n=20 时,10 条指令如下: 操作 回答 操作后的数组 1 1 10 N/A 11111111110000000000 2 6 1 11111111110000000000 2 12 0 11111111110000000000 1 5 12 N/A 11110000001100

「一本通 4.2 例 2」最敏捷的机器人(loj10120)

题目描述 Wind 设计了很多机器人.但是它们都认为自己是最强的,于是,一场比赛开始了-- 机器人们都想知道谁是最敏捷的,于是它们进行了如下一个比赛.首先,他们面前会有一排共 n 个数,它们比赛看谁能最先把每连续 k 个数中最大和最小值写下来,当然,这些机器人运算速度都很快,它们比赛的是谁写得快. 但是 Wind 也想知道答案,你能帮助他吗? 输入格式 第一行为 n,k,意义如题目描述. 第二行共 n 个数,为数字序列,所有数字均在 Pascal 的 longint 范围内,即所有数均为整数,且

loj10139. 「一本通 4.5 练习 1」树上操作(loj2125. 「HAOI2015」树上操作 )

思路: 第二遍dfs时记录end[x]为在结点序列中以x为根的子树最后访问的节点,写个线段树标记下传即可.与值有关的数据注意long long #include<cstdio> #include<iostream> #include<vector> #include<cmath> #include<string> using namespace std; const int maxn = 100010; inline void qread(int

loj10143. 「一本通 4.6 例 1」营业额统计

思路: 使用treap在线存数据,每次取出新读入数据的前驱与后继,与该数据差值较小的就是.(注意若已有同一个数据,特判ans+=0) #include<cstdio> #include<iostream> #include<cstdlib> #include<cmath> using namespace std; const int maxn = 100010; inline void qread(int &x) { x = 0; register

loj10144. 「一本通 4.6 练习 1」宠物收养所

思路: treap+简单模拟. 我只能说:注意取模!!!! #include<cstdio> #include<iostream> #include<cstdlib> #include<cmath> using namespace std; const int maxn = 80010; inline void qread(int &x) { x = 0; register int ch = getchar(), flag = 0; while(ch

loj10145. 「一本通 4.6 练习 2」郁闷的出纳员

思路: 简单的treap+模拟. 但这道题的题面有误导群众的嫌疑,因为题中并没有明确刚进入公司就离开的算不算在离开公司的人中. 当然,对于这道题的数据,题目意思是不算在内. #include<cstdio> #include<iostream> #include<cstdlib> #include<string> using namespace std; const int maxn = 80010; inline void qread(int &x

loj10147. 「一本通 5.1 例 1」石子合并

思路: 经典的区间dp,题解到处都是... #include<cstdio> #include<iostream> #include<cstdio> using namespace std; const int maxn = 1010; inline void qread(int &x){ x = 0; register int ch = getchar(); while(ch < '0' || ch > '9') ch = getchar(); w

loj10159. 「一本通 5.2 练习 2」旅游规划

思路: 树形dp,最后用dfs标记所有的路径上的点,要注意可能会有多个点在以它为根的子树中的最长路径等于整棵树的最长路径,要分别作为初始点遍历整棵树. #include<cstdio> #include<iostream> #include<cstdlib> #include<vector> using namespace std; const int maxn = 200010; void qread(int &x) { x = 0; regist