bzoj 4999: This Problem Is Too Simple!

题意:正常人都看得懂吧
给您一颗树,每个节点有个初始值。
现在支持以下两种操作:
1. C i x(0<=x<2^31) 表示将i节点的值改为x。
2. Q i j x(0<=x<2^31) 表示询问i节点到j节点的路径上有多少个值为x的节点。
题解:
树链剖分+线段树。先把所有值离散化,然后开许多棵线段树,记录每个值的个数。因为空间过不去,所以我用了许多指针。真tm不习惯
代码:

#include<bits/stdc++.h>
using namespace std;

struct tree
{
    int l,r,c;
    tree *lc,*rc;
    tree(int _l,int _r)
    {
        l=_l;
        r=_r;
        c=0;
        lc=rc=NULL;
    }
}*root[300010];
struct pnt
{
    int f,s,sz,en,tp,dp;
}p[100010];
struct edge
{
    int x,y,n;
}e[200010];
struct query
{
    char s[2];
    int x,y,z;
}q[200010];
int n,m,nn=0,a[100010],fst[100010],num=0;
map<int,int>lsh;

void ins(int x,int y)
{
    e[++num]={x,y,fst[x]};
    fst[x]=num;
}
void dfs1(int x,int f)
{
    p[x].f=f;
    p[x].dp=p[f].dp+1;
    p[x].sz=1;
    for(int i=fst[x];i;i=e[i].n)
    {
        int y=e[i].y;
        if(y==f)
        continue;
        dfs1(y,x);
        if(p[y].sz>p[p[x].s].sz)
        p[x].s=y;
        p[x].sz+=p[y].sz;
    }
}
void dfs2(int x,int tp)
{
    p[x].tp=tp;
    p[x].en=++nn;
    if(p[x].s)
    {
        dfs2(p[x].s,tp);
        for(int i=fst[x];i;i=e[i].n)
        {
            int y=e[i].y;
            if(y==p[x].f||y==p[x].s)
            continue;
            dfs2(y,y);
        }
    }
}
bool del(tree* i,int x)
{
    if(i==NULL)
    return 1;
//  printf("del %d %d %d\n",i->l,i->r,x);
    if(i->l==i->r)
    {
        delete i;
        return 1;
    }
    int md=i->l+i->r>>1;
    if(x<=md)
    {
        if(del(i->lc,x))
        i->lc=NULL;
    }
    else
    {
        if(del(i->rc,x))
        i->rc=NULL;
    }
//  i->c=(i->lc!=NULL?i->lc->c:0)+(i->rc!=NULL?i->rc->c:0);
    i->c--;
    if(i->c==0)
    {
        delete i;
        return 1;
    }
    return 0;
}
void add(tree* i,int x)
{
//  printf("add %d %d %d\n",i->l,i->r,x);
//  system("pause");
    if(i->l==i->r)
    {
        i->c=1;
        return;
    }
    int md=i->l+i->r>>1;
    tree *lc=i->lc,*rc=i->rc;
    if(x<=md)
    {
        if(lc==NULL)
        i->lc=lc=new tree(i->l,md);
        add(lc,x);
    }
    else
    {
        if(rc==NULL)
        i->rc=rc=new tree(md+1,i->r);
        add(rc,x);
    }
    i->c=(lc!=NULL?lc->c:0)+(rc!=NULL?rc->c:0);
}
int get(tree* i,int l,int r)
{
    if(i->l==l&&i->r==r)
    {
        return i->c;
    }
    int md=i->l+i->r>>1;
    tree *lc=i->lc,*rc=i->rc;
    if(r<=md)
    {
        return lc==NULL?0:get(lc,l,r);
    }
    else if(l>md)
    {
        return rc==NULL?0:get(rc,l,r);
    }
    else
    {
        return (lc==NULL?0:get(lc,l,md))+(rc==NULL?0:get(rc,md+1,r));
    }
}
int slv(int x,int y,int c)
{
    int ans=0;
    while(p[x].tp!=p[y].tp)
    {
        if(p[p[x].tp].dp<p[p[y].tp].dp)
        swap(x,y);
        int tx=p[x].tp,ty=p[y].tp;
        ans+=get(root[c],p[tx].en,p[x].en);
        x=p[tx].f;
    }
    if(p[x].dp<p[y].dp)
    swap(x,y);
    ans+=get(root[c],p[y].en,p[x].en);
    return ans;
}
void print(tree *x)
{
    if(x==NULL)
    return;
    printf("l:%d r:%d c:%d\n",x->l,x->r,x->c);
    print(x->lc);
    print(x->rc);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        lsh[a[i]]=1;
    }
    for(int i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        ins(x,y);
        ins(y,x);
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%s%d%d",q[i].s,&q[i].x,&q[i].y);
        if(q[i].s[0]=='Q')
        {
            scanf("%d",&q[i].z);
            lsh[q[i].z]=1;
        }
        else
        lsh[q[i].y]=1;
    }
    int cnt=0;
    for(map<int,int>::iterator it=lsh.begin();it!=lsh.end();it++)
    {
        it->second=cnt++;
//      printf("lsh[%d]=%d\n",it->first,it->second);
    }
    dfs1(1,0);
    dfs2(1,1);/*
    puts("en:");
    for(int i=1;i<=n;i++)
    printf("%d ",p[i].en);
    puts("");*/
    for(int i=0;i<cnt;i++)
    root[i]=new tree(1,nn);
    for(int i=1;i<=n;i++)
    {
        a[i]=lsh[a[i]];
        add(root[a[i]],p[i].en);
    }/*
        for(int i=0;i<cnt;i++)
        {
            printf("tree:%d\n",i);
            print(root[i]);
        }*/
    for(int i=1;i<=m;i++)
    {
        if(q[i].s[0]=='C')
        {
            q[i].y=lsh[q[i].y];
            if(del(root[a[q[i].x]],p[q[i].x].en))
            root[a[q[i].x]]=new tree(1,nn);
            add(root[q[i].y],p[q[i].x].en);
            a[q[i].x]=q[i].y;
        }
        else
        {
            q[i].z=lsh[q[i].z];
            printf("%d\n",slv(q[i].x,q[i].y,q[i].z));
        }/*
        for(int i=0;i<cnt;i++)
        {
            printf("tree:%d\n",i);
            print(root[i]);
        }*/
    }
}
发布了99 篇原创文章 · 获赞 17 · 访问量 1万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 编程工作室 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览