zoukankan      html  css  js  c++  java
  • J

    题意:给一颗树,两种操作,查询 i 结点的颜色,和将i结点和它的子树都染成另一种颜色

    题解:dfs序构建线段树,对于x和其子树染色就是 l[x] 和 r[x];

    dfs序线段树板子

    #include<iostream>
    #include<cstdio>
    #include<cstdlib>
    #include<cstring>
    #include<algorithm>
    #include<cmath>
    #include<queue>
    #include<list>
    #include<math.h>
    #include<vector>
    #include<stack>
    #include<string>
    #include<stdio.h>
    
    using namespace std;
    typedef long long LL;
    const int MAXN = 1e5 + 10;
    
    vector<int>vec[MAXN];
    int cnt[MAXN],l[MAXN],r[MAXN],vis[MAXN];
    int st[MAXN << 2];
    void init() {
        memset(l,-1,sizeof l);
        memset(r, -1 ,sizeof r);
        memset(cnt, 0, sizeof cnt);
        memset(vis, 0, sizeof vis);
    }
    
    void dfs(int x,int &pos) {
        l[x] = pos ++;
        for(int i = 0; i < vec[x].size(); i++) {
            dfs(vec[x][i],pos);
        }
        r[x] = pos++;
    }
    void build(int o,int l,int r) {
        st[o] = -1;
        if(l == r) return;
        int mid = (l + r) >> 1;
        build(o << 1, l , mid);
        build(o << 1 | 1, mid + 1, r);
    }
    void pushdown(int o) {
        if(st[o] != -1) {
            st[o << 1] = st[o];
            st[o << 1 | 1 ] = st[o];
            st[o] = -1;
        }
    }
    int query(int o,int l,int r,int x) {
        if(l == r) return st[o];
        pushdown(o);
        int mid = (l + r) >> 1;
        if(x <= mid) query(o << 1, l, mid, x);
        else query(o << 1 | 1, mid + 1, r, x);
    }
    void update(int o,int l,int r,int ql,int qr,int y) {
        if(ql <= l && r  <= qr) {
            st[o] = y;
            return;
        }
        pushdown(o);
        int mid = (l + r) >> 1;
        if(ql <= mid) update(o << 1, l, mid, ql, qr, y);
        if(qr >= mid + 1) update(o << 1 | 1, mid + 1, r, ql, qr, y);
    }
    int main() {
        int t, _ = 1;
        scanf("%d", &t);
        while (t--) {
            init();
            printf("Case #%d:
    ", _++);
            int n;
            scanf("%d", &n);
            for(int i = 1; i <= n; i++)
                vec[i].clear();
            build(1, 1, 2 * n);
            int u, v;
            for (int i = 1; i < n; i++) {
                scanf("%d %d", &u, &v);
                vec[v].push_back(u);
                cnt[u]++;
            }
            int pos = 1;
            for (int i = 1; i <= n; i++) {
                if (cnt[i] == 0) {
                    dfs(i,pos);
                    break;
                }
            }
    //        for(int i = 1;i<=n;i++)
    //            printf("%d %d %d
    ",i,l[i],r[i]);
    
            int m;
            scanf("%d",&m);
            while (m--) {
                char s[5];
                int x,y;
                scanf("%s",s);
                if(s[0] == 'C') {
                    scanf("%d",&x);
                    printf("%d
    ",query(1,1,2 * n,l[x]));
                } else {
                    scanf("%d %d",&x,&y);
                    update(1, 1, 2 * n, l[x],r[x], y);
                }
            }
        }
    }
    
    /*
    
    1
    5
    4 3
    3 2
    1 3
    5 2
    5
    C 3
    T 2 1
    C 3
    T 3 2
    C 3
    
    
     */
  • 相关阅读:
    截取某一个元素的图
    11、python的异常机制
    10、python面向对象三大特性:封装、继承、多态
    9、python之面向对象
    软件测试有前途吗?
    对应届生做测试的同学们的一些建议
    没有代码基础如何学习自动化测试?
    接口自动化测试有哪些工具或者框架?
    软件测试流程
    接口自动化测试中logging实际用法
  • 原文地址:https://www.cnblogs.com/smallhester/p/11334761.html
Copyright © 2011-2022 走看看