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)log2m)
……
……
……
其实还有一种想法
沿用前面对每种颜色建立邻接表的思路
这次我们记录前驱节点
每次查询[l,r]就是看[l,r]中前驱节点小于l的数量
(其实也可以建立后继节点,查询大于r的数量,不过那样要处理每种颜色最后出现的位置的后继blabla)
显然这是主席树可以做的事
复杂度是
O(mlog2m)
主要是排序时间,查询是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();
}