【HDU5156】Harry and Christmas tree，两种离线的做法

Time:2016.08.11
Author:xiaoyimi
转载注明出处谢谢

---

思路：
前天的模拟赛题目
来自BC#25
这种修改与查询分开的情况优先考虑离线做法
经典的颜色统计，移植到树上
每次查询就是求dfs序[l,r]
具体思路如下

求出dfs序后以此为关键字对修改排序
对每种颜色建一个类似邻接表（？？），可以顺序查找的东西
对每种颜色，每次修改都记录后继节点（这里记录的后继节点是dfs序）
将查询按左端点dfs序为关键字排序
当前修改到的序号now=1
每一次的查询时看一下当前now操作所修改节点的dfs序是否小于当前查询的左端点dfs序
是则依次如下操作
1.now操作的后继节点若不为空，则把后继节点在树状数组中+1
2.now++
不是则退出
每个询问的答案就是右端点的前缀和减去左端点-1的前缀和
因为询问的左端点和操作的dfs序都是递增的
复杂度$O((n+m)log_2m)$
……
……
……
其实还有一种想法
沿用前面对每种颜色建立邻接表的思路
这次我们记录前驱节点
每次查询[l,r]就是看[l,r]中前驱节点小于l的数量
（其实也可以建立后继节点，查询大于r的数量，不过那样要处理每种颜色最后出现的位置的后继blabla）
显然这是主席树可以做的事
复杂度是$O(mlog_2m)$主要是排序时间，查询是nlogn的
但空间复杂度是$O(mlogn)~O(mlogm)$
64M开不下……
128M就好了……
注意：
ヾ(｡｀Д´｡)卡常数，之前用个vectorT飞了，2.5s的题跑了2.3~2.4s真是个笑话
代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define M 500005
#define N 50005
#define low(i) i&(-i)
using namespace std;
int n,m,tot,cnt;
int first[N],last[N<<1],pre[M],ans[N],d[N],L[N];
struct edge{
    int v,next;
}e[N<<1];
struct node{
    int x,c;
    bool operator <(node other)const{return L[x]<L[other.x];}
}a[M];
struct query{
    int id,l,r;
    bool operator <(query other)const{return l<other.l;}
}q[N];
int in()
{
    int t=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') t=(t<<3)+(t<<1)+ch-48,ch=getchar();
    return t;
}
void add(int x,int y)
{
    e[++tot].v=y;e[tot].next=first[x];first[x]=tot;
    e[++tot].v=x;e[tot].next=first[y];first[y]=tot;
}
void dfs(int x,int fa)
{
    q[x].id=x;
    L[x]=q[x].l=++cnt;
    for (int i=first[x];i;i=e[i].next)
        if (fa!=e[i].v)
            dfs(e[i].v,x);
    q[x].r=cnt;
}
void update(int x,int val)
{
    for (;x<=n;x+=low(x))
        d[x]+=val; 
}
int get(int x)
{
    int t=0;
    for (;x;x-=low(x))
        t+=d[x];
    return t;
}
void work()
{
    memset(last,0,sizeof(last));
    cnt=tot=0;
    for (int i=1;i<n;i++) add(in(),in());
    dfs(1,0);
    for (int i=1;i<=m;i++)
        a[i].x=in(),
        a[i].c=in();
    sort(a+1,a+m+1);
    for (int i=m;i;i--)
        pre[i]=last[a[i].c],
        last[a[i].c]=L[a[i].x];
    sort(q+1,q+n+1);
    int now=1;
    for (int i=1;i<=100000;i++)
        if (last[i]) update(last[i],1);
    for (int i=1;i<=n;i++)
    {
        while (q[i].l>L[a[now].x]&&now<=m)
        {
            if (pre[now]) update(pre[now],1);
            now++;
        } 
        ans[q[i].id]=get(q[i].r)-get(q[i].l-1);
    }
    for (int i=1;i<=n;i++)
        first[i]=d[i]=0,
        printf("%d%c",ans[i]," \n"[i==n]);
}
int main()
{
    while (~scanf("%d%d",&n,&m)) work();
}