[codeforces375D]Tree and Queries（dsu on the tree+bit）

最新推荐文章于 2020-11-11 13:58:30 发布

wwyx2001

最新推荐文章于 2020-11-11 13:58:30 发布

阅读量432

点赞数 1

CC 4.0 BY-SA版权

分类专栏：树状数组 dsu on tree

本文链接：https://blog.youkuaiyun.com/Blue_CuSO4/article/details/78175034

树状数组同时被 2 个专栏收录

15 篇文章

订阅专栏

dsu on tree

5 篇文章

订阅专栏

本文介绍了一种解决特定树形结构问题的高效算法——DSU on Tree，并通过实例讲解了如何利用该算法统计子树中颜色出现次数大于等于k的情况。文章详细展示了算法的具体实现过程，包括数据结构设计与关键函数的编写。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：

我是超链接

题意：

统计以v为顶点的子树中颜色出现大于等于k次的种类数

题解：

dsu on the tree！
先把询问记录下来—-这里的询问记录很多都用虚树呢
这里因为是至少k次，用bit就很好了，这里对于bit的处理十分巧妙，记录出现多少次的颜色种类

add操作
if (cnt[a[x]]) change(cnt[a[x]],-1);
cnt[a[x]]+=vv;
if (cnt[a[x]]) change(cnt[a[x]],1);

没有严格的父子关系就不要连单向边啊

代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#define LL long long
#define N 100005 
#define inf 1e5
using namespace std;
int tot,nxt[N*2],point[N],v[N*2],size[N],tot1,nxt1[N*2],point1[N],v1[N*2],id[N],h[N],son[N];
int cnt[N],Son,c[N],a[N],ans[N];
void addline(int x,int y){
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x;
}
void addline1(int x,int y,int i){++tot1; nxt1[tot1]=point1[x]; point1[x]=tot1; v1[tot1]=y; id[tot1]=i;}
void getson(int x,int fa)
{
    size[x]=1;
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa)
      {
        getson(v[i],x);
        if (size[v[i]]>size[son[x]]) son[x]=v[i];
        size[x]+=size[v[i]];
      }
}
void change(int loc,int vv)
{
    for (int i=loc;i<=inf;i+=i&(-i))
      c[i]+=vv;
}
int ask(int loc)
{
    int ans=0;
    for (int i=loc;i>=1;i-=i&(-i))
      ans+=c[i];
    return ans;
}
void add(int x,int fa,int vv)
{
    if (cnt[a[x]]) change(cnt[a[x]],-1);
    cnt[a[x]]+=vv;
    if (cnt[a[x]]) change(cnt[a[x]],1);
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa && v[i]!=Son) add(v[i],x,vv);
}
void dfs(int x,int fa,int k)
{
    for (int i=point[x];i;i=nxt[i])
      if (v[i]!=fa && v[i]!=son[x]) dfs(v[i],x,0); 
    if (son[x]) dfs(son[x],x,1),Son=son[x];
    add(x,fa,1);   Son=0;
    for (int i=point1[x];i;i=nxt1[i])
    {
        int xx=ask(inf),yy=ask(v1[i]-1);
        ans[id[i]]=xx-yy; 
    }
    if (!k) add(x,fa,-1);
}
int main()
{
    int n,i,m;
    scanf("%d%d",&n,&m);
    for (i=1;i<=n;i++) scanf("%d",&a[i]);
    for (i=1;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        addline(x,y);
    }
    getson(1,0);
    for (i=1;i<=m;i++)
    {
        int vv,hh;
        scanf("%d%d",&vv,&hh); 
        addline1(vv,hh,i);
    }
    dfs(1,0,0);
    for (i=1;i<=m;i++) printf("%d\n",ans[i]);
}