CSUOJ 2354 tree 树链剖分

http://192.168.1.212:20080/csuoj/problemset/problem?pid=2354
Description
可怜的史蒂夫在暑假里又迷上了一款新的游戏，游戏里面有成百上千个国家，并分别由不同的玩家管理着。史蒂夫管理的国家由n个城市组成，编号依次为1-n, 其中1号城市为首都，与此同时，这些城市由n-1条道路连接，使得任意两个城市之间可以相互到达。现在有m个敌对国想要占领史蒂夫的国家，每个敌对国在进攻开始前会选择k个不同的城市空降军队，然后依次从这k个城市出发，沿着最短路径向首都1号城市发动k次进攻，对于每个经过的城市，敌对国军队都会在这个城市里插上自己的国旗。在所有进攻完成后，史蒂夫想知道每个城市被插上了多少面不同的旗子。

Input
输入第一行为一个整数T（T<=50)，表示数据组数。

对于每组数据，第一行三个整数n,m,k，(1<=n<=20000,1<=m<=5000,1<=k<=min(20,n))分别表示城市个数，敌对国个数，每个敌对国发动进攻的次数。接下来n-1行，每行两个整数，分别表示n-1条道路所连接的城市编号。最后m行，每行k个整数，分别表示这m个敌对国发动k次进攻的起始城市。

m*k>10000的数据组数不超过5组。

Output
对于每组数据，输出n行，每行一个整数，第i个数表示所有进攻完成后，第i个城市被插上的不同旗子数。

Sample Input
1

5 5 2
1 2
2 3
3 4
2 5
2 1
3 4
3 4
5 2
4 5
Sample Output
5
5
3
3
2
Hint
敌对国1经过的城市为 1，2

敌对国2经过的城市为 1，2，3，4

敌对国3经过的城市为 1，2，3，4

敌对国4经过的城市为 1，2，5

敌对国5经过的城市为 1，2，3，4，5

Source
Author
lts
思路：树链剖分。关键是怎么去重，考虑记录所有的区间，对区间排序后进行合并操作，得到了一个个不相交的区间。此时再进行更新就不会重复了。

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<cstring>
#define pr pair<int,int>
using namespace std;

const int maxn=2e5+5;

struct edge
{
    int to,nxt;
}Edge[maxn<<1];

struct node
{
    int l,r;
    int sum,lazy;
}tree[maxn<<2];

vector<pr> vec;
int siz[maxn],son[maxn],fa[maxn],top[maxn],deep[maxn],pos[maxn],head[maxn];
int ans[maxn];
int n,m,k,tot,cnt;

inline void addedge(int u,int v)
{
    Edge[++cnt].to=v,Edge[cnt].nxt=head[u],head[u]=cnt;
}

void dfs1(int u,int f)
{
    siz[u]=1,son[u]=0,fa[u]=f,deep[u]=deep[f]+1;
    int to;
    for(int i=head[u];i;i=Edge[i].nxt)
    {
        to=Edge[i].to;
        if(to!=f)
        {
            dfs1(to,u);
            siz[u]+=siz[to];
            if(siz[son[u]]<siz[to])
                son[u]=to;
        }
    }
}

void dfs2(int u,int f,int k)
{
    top[u]=k,pos[u]=++tot;
    if(son[u])
        dfs2(son[u],u,k);
    int to;
    for(int i=head[u];i;i=Edge[i].nxt)
    {
        to=Edge[i].to;
        if(to!=f&&to!=son[u])
            dfs2(to,u,to);
    }
}

void build(int i,int l,int r)
{
    tree[i].l=l,tree[i].r=r;
    tree[i].lazy=tree[i].sum=0;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
}

void down(int i)
{
    tree[i<<1].lazy+=tree[i].lazy;
    tree[i<<1|1].lazy+=tree[i].lazy;
    tree[i<<1].sum+=tree[i].lazy;
    tree[i<<1|1].sum+=tree[i].lazy;
    tree[i].lazy=0;
}

void update(int i,int l,int r)
{
    if(tree[i].l==l&&tree[i].r==r)
    {
        tree[i].lazy+=1;
        tree[i].sum+=1;
        return ;
    }
    if(tree[i].lazy)
        down(i);
    int mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid)
        update(i<<1,l,r);
    else if(l>mid)
        update(i<<1|1,l,r);
    else
        update(i<<1,l,mid),
        update(i<<1|1,mid+1,r);
}

int query(int i,int pos)
{
    if(tree[i].l==tree[i].r)
        return tree[i].sum;
    if(tree[i].lazy)
        down(i);
    int mid=(tree[i].l+tree[i].r)>>1;
    if(pos<=mid)
        return query(i<<1,pos);
    else
        return query(i<<1|1,pos);
}

void work(int v)
{
    int u=1;
    while(top[u]!=top[v])
    {
        if(deep[top[u]]<deep[top[v]])
            swap(u,v);
        vec.push_back(pr(pos[top[u]],pos[u]));
        u=fa[top[u]];
    }
    if(deep[u]>deep[v])
        swap(u,v);
    vec.push_back(pr(pos[u],pos[v]));
}

void solve()
{
    sort(vec.begin(),vec.end());
    int len=vec.size();
    int prel=vec[0].first;
    int prer=vec[0].second;
    for(int i=1;i<len;i++)
    {
        if(vec[i].first<=prer)
            prer=max(prer,vec[i].second);
        else
        {
           // cout<<prel<<'!'<<prer<<endl;
            update(1,prel,prer);
            prel=vec[i].first;
            prer=vec[i].second;
        }
    }
  //  cout<<prel<<'!'<<prer<<endl;
    update(1,prel,prer);
    vec.clear();
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        tot=cnt=0;
        memset(head,0,sizeof(head));
        scanf("%d%d%d",&n,&m,&k);
        int x,y;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&x,&y);
            addedge(x,y),addedge(y,x);
        }
        dfs1(1,0);
        dfs2(1,0,1);
        build(1,1,n);
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<k;j++)
            {
                scanf("%d",&x);
                work(x);
            }
            solve();
        }
        for(int i=1;i<=n;i++)
			printf("%d\n",query(1,pos[i]));
    }
    return 0;
}