JZOJsenior4005.【GDKOI2015】树

最新推荐文章于 2019-03-26 18:25:22 发布

原创最新推荐文章于 2019-03-26 18:25:22 发布 · 380 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#模拟赛 #GDKOI #树链剖分 #暴力

模拟赛同时被 3 个专栏收录

240 篇文章

订阅专栏

暴力

25 篇文章

订阅专栏

树链剖分

9 篇文章

订阅专栏

本文介绍了一种结合树剖与线段树的数据结构应用案例，通过具体代码实现了解决特定查询问题的方法。该方法利用树剖进行预处理，并通过维护10棵线段树来高效地更新节点颜色及权值，适用于处理复杂的数据查询与更新操作。

problem

Description

Input

Output

对于每一次询问，输出一行。

Sample Input

样例输入1：
3
0 1 2
0 1 2
0 1
0 2
4
Ask 1 2 0
Change 0 1 2
Set 0 1 98
Ask 1 2 1

样例输入2：
3
0 1 2
0 1 2
0 1
0 2
4
Ask 1 2 1
Change 0 1 2
Set 0 1 98
Ask 1 2 2

Sample Output

样例输入1：
0
100

样例输出2：
1
1

Data Constraint

analysis

正解大树剖
对于剖好的树，建10棵线段树来维护颜色和权值
改变颜色就打上很奇怪的lazy-tag
cgh树剖切掉打了11000+bytes~~（树剖之神）~~
~~我全部操作打了暴力（连树上倍增都不打）就过了~~
~~数据水怪我咯~~

code

#include<stdio.h>
#define MAXN 50001
#define MAXM 100001

using namespace std;

int last[MAXM],next[MAXM],tov[MAXM];
int fa[MAXN],depth[MAXN];
int n,m,tot;
char s[10];

struct node
{
    int c,v;
}a[MAXN];

void insert(int x,int y)
{
    next[++tot]=last[x];
    last[x]=tot;
    tov[tot]=y;
}

void dfs(int x)
{
    for (int i=last[x];i;i=next[i])
    {
        int j=tov[i];
        if (j!=fa[x])
        {
            fa[j]=x;
            depth[j]=depth[x]+1;
            dfs(j);
        }
    }
}

int lca(int x,int y,int z)
{
    int ans=0;
    while (x!=y)
    {
        if (depth[x]>depth[y])
        {
            if (a[x].c==z)ans+=a[x].v;
            x=fa[x];
        }
        else 
        {
            if (a[y].c==z)ans+=a[y].v;
            y=fa[y];
        }       
    }
    if (a[x].c==z)ans+=a[x].v;
    return ans;
}

void change(int x,int y,int z)
{
    if (a[x].c==y)
    {
        a[x].c=z;   
    }
    else if (a[x].c==z)
    {
        a[x].c=y;
    }
    for (int i=last[x];i;i=next[i])
    {
        int j=tov[i];
        if (j!=fa[x])
        {
            change(j,y,z);
        }
    }
}

int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;i++)scanf("%d",&a[i].c);
    for (int i=0;i<n;i++)scanf("%d",&a[i].v);
    for (int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        insert(x,y);
        insert(y,x);
    }
    depth[0]=1;
    dfs(0);
    scanf("%d",&m);
    while (m--)
    {
        int x,y,z;
        scanf("%s%d%d%d",&s,&x,&y,&z);
        if (s[0]=='A')printf("%d\n",lca(x,y,z));
        else if (s[0]=='C')change(x,y,z);
        else if (s[0]=='S')a[x].c=y,a[x].v=z;
    }
    return 0;
}