dsu on tree

题目链接

点我跳转

题目大意

给定一棵 \(n\) 个节点的树,根节点为 \(1\)。每个节点上有一个颜色 \(c_i\)

\(m\) 次询问。

每次询问给出 \(u\) \(k\):询问在以 \(u\) 为根的子树中,出现次数 \(≥k\) 的颜色有多少种。

解题思路

可以开棵权值线段树

如果当前颜色出现的次数 \(cnt[i] = x\), 就把树的第 \(x\) 个位置的值 \(+ 1\)

那么对于每个询问的 \(k\) 输出树的第 \(k\) 个位置的值即可

AC_Code

#include<bits/stdc++.h>
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,n,a) for (int i=n;i>=a;i--)
#define int long long
#define pb push_back
#define fi first
#define se second
using namespace std;
const int N = 3e5 + 10;
struct Tree{
int l , r , lazy , sum;
}tree[N << 2];
void push_up(int rt)
{
tree[rt].sum = tree[rt << 1].sum + tree[rt << 1 | 1].sum;
}
void push_down(int rt)
{
int x = tree[rt].lazy;
tree[rt].lazy = 0;
tree[rt << 1].lazy = tree[rt << 1 | 1].lazy = x;
tree[rt << 1].sum += (tree[rt << 1].r - tree[rt << 1].l + 1) * x;
tree[rt << 1 | 1].sum += (tree[rt << 1 | 1].r - tree[rt << 1 | 1].l + 1) * x;
}
void build(int l , int r , int rt)
{
tree[rt].l = l , tree[rt].r = r , tree[rt].lazy = 0;
if(l == r)
{
tree[rt].sum = 0;
return ;
}
int mid = l + r >> 1;
build(l , mid , rt << 1);
build(mid + 1 , r , rt << 1 | 1);
push_up(rt);
}
void update_range(int L , int R , int rt , int val)
{
int l = tree[rt].l , r = tree[rt].r;
if(L <= l && r <= R)
{
tree[rt].lazy += val;
tree[rt].sum += (r - l + 1) * val;
return ;
}
push_down(rt);
int mid = l + r >> 1;
if(L <= mid) update_range(L , R , rt << 1 , val);
if(R > mid) update_range(L , R , rt << 1 | 1 , val);
push_up(rt);
}
int query_range(int L , int R , int rt)
{
int l = tree[rt].l , r = tree[rt].r;
if(L <= l && r <= R) return tree[rt].sum;
push_down(rt);
int mid = l + r >> 1 , ans = 0;
if(L <= mid) ans += query_range(L , R , rt << 1);
if(R > mid) ans += query_range(L , R , rt << 1 | 1);
return ans;
}
struct Edge{
int nex , to;
}edge[N << 1];
int head[N] , TOT;
void add_edge(int u , int v)
{
edge[++ TOT].nex = head[u] ;
edge[TOT].to = v;
head[u] = TOT;
}
int dep[N] , sz[N] , hson[N] , HH;
int col[N] , n , m , up;
int cnt[N] , sum[N];
vector<pair<int , int>>Q[N] , ans;
void dfs(int u , int far)
{
dep[u] = dep[far] + 1;
sz[u] = 1;
for(int i = head[u] ; i ; i = edge[i].nex)
{
int v = edge[i].to;
if(v == far) continue ;
dfs(v , u);
sz[u] += sz[v];
if(sz[v] > sz[hson[u]]) hson[u] = v;
}
}
void calc(int u , int far, int val)
{
cnt[col[u]] += val;
if(val == 1)
{
int k = cnt[col[u]];
update_range(k , k , 1 , 1);
}
if(val == -1)
{
int k = cnt[col[u]] + 1;
update_range(k , k , 1 , -1);
}
for(int i = head[u] ; i ; i = edge[i].nex)
{
int v = edge[i].to;
if(v == far || v == HH) continue ;
calc(v , u , val);
}
}
void dsu(int u , int far , int op)
{
for(int i = head[u] ; i ; i = edge[i].nex)
{
int v = edge[i].to;
if(v == far || v == hson[u]) continue ;
dsu(v , u , 0);
}
if(hson[u]) dsu(hson[u] , u , 1) , HH = hson[u];
calc(u , far , 1);
for(auto i : Q[u])
{
int id = i.fi , k = i.se;
int res = query_range(k , k , 1);
ans.pb(make_pair(id , res));
}
HH = 0;
if(!op) calc(u , far , -1);
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0) , cout.tie(0);
cin >> n >> m;
rep(i , 1 , n) cin >> col[i];
rep(i , 1 , n - 1)
{
int u , v;
cin >> u >> v;
add_edge(u , v) , add_edge(v , u);
}
rep(i , 1 , m)
{
int u , k;
cin >> u >> k;
Q[u].pb(make_pair(i , k));
}
build(1 , 100000 , 1);
dfs(1 , 0);
dsu(1 , 0 , 0);
sort(ans.begin() , ans.end());
for(auto i : ans) cout << i.se << '\n';
return 0;
}

Codeforces375D Tree and Queries的更多相关文章

  1. [Codeforces375D]Tree and Queries(莫队算法)

    题意:给定一棵树,每个节点有颜色,对于每个询问(u,k)询问以u为根节点的子树下有多少种颜色出现次数>=k 因为是子树,跟dfs序有关,转化为一段区间,可以用莫队算法求解 直接用一个数组统计出现 ...

  2. [Codeforces Round #221 (Div. 1)][D. Tree and Queries]

    题目链接:375D - Tree and Queries 题目大意:给你一个有n个点的树,每个点都有其对应的颜色,给出m次询问(v,k),问v的子树中有多少种颜色至少出现k次 题解:先对所有的询问进行 ...

  3. Codeforces 375D Tree and Queries(DFS序+莫队+树状数组)

    题目链接  Tree and Queries 题目大意  给出一棵树和每个节点的颜色.每次询问$vj, kj$ 你需要回答在以$vj$为根的子树中满足条件的的颜色数目, 条件:具有该颜色的节点数量至少 ...

  4. CodeForces 375D Tree and Queries 莫队||DFS序

    Tree and Queries 题意:有一颗以1号节点为根的树,每一个节点有一个自己的颜色,求出节点v的子数上颜色出现次数>=k的颜色种类. 题解:使用莫队处理这个问题,将树转变成DFS序区间 ...

  5. codeforces 375D:Tree and Queries

    Description You have a rooted tree consisting of n vertices. Each vertex of the tree has some color. ...

  6. CF375D Tree and Queries

    题意翻译 给出一棵 n 个结点的树,每个结点有一个颜色 c i . 询问 q 次,每次询问以 v 结点为根的子树中,出现次数 ≥k 的颜色有多少种.树的根节点是1. 感谢@elijahqi 提供的翻译 ...

  7. CodeForces 376F Tree and Queries(假·树上莫队)

    You have a rooted tree consisting of n vertices. Each vertex of the tree has some color. We will ass ...

  8. Codeforces 375 D Tree and Queries

    Discription You have a rooted tree consisting of n vertices. Each vertex of the tree has some color. ...

  9. CodeForces - 375D Tree and Queries (莫队+dfs序+树状数组)

    You have a rooted tree consisting of n vertices. Each vertex of the tree has some color. We will ass ...

随机推荐

  1. Parcelable使用(一)

    android有两种序列化方式:一是实现Serializable接口(是JavaSE本身就支持的),二是实现Parcelable接口(是Android特有功能,效率比实现Serializable接口高 ...

  2. eclipse时一直卡在进程中

    (1)今天遇到进入eclipse时一直卡在 进程中,无论是重启电脑,还是重启软件 删除 D:\workspace\.metadata\.lock 文件才有用,特此记录下. (2)还有一种情况就是打开e ...

  3. D. Equalize the Remainders 解析(思維)

    Codeforce 999 D. Equalize the Remainders 解析(思維) 今天我們來看看CF999D 題目連結 題目 略,請直接看原題 前言 感覺要搞個類似\(stack\)的東 ...

  4. Win32之创建进程

    CreateProcess函数介绍 BOOL CreateProcessA( LPCSTR lpApplicationName, //可执行文件的名称完整的路径+程序名字) LPSTR lpComma ...

  5. 签到功能,用 MySQL 还是 Redis ?

    现在的网站和app开发中,签到是一个很常见的功能,如微博签到送积分,签到排行榜. 如移动app ,签到送流量等活动.   用户签到是提高用户粘性的有效手段,用的好能事半功倍! 下面我们从技术方面看看常 ...

  6. linux下生成动态库和链接动态库

    1.生成动态库 src/test.h #ifndef _TEST_H_HH #define _TEST_H_HH void print(); #endif src/test.cpp #include ...

  7. Centos7中MySQL8.0安装过程及遇到的问题记录

    本文主要参考以下2篇文章,安装操作 https://www.miboxapp.com/article/detail/1146659339214393344 https://blog.csdn.net/ ...

  8. Django项目登录注册系统

    Django项目之个人网站 关注公众号"轻松学编程"了解更多. Github地址:https://github.com/liangdongchang/MyWeb.git 感兴趣的可 ...

  9. 如何k个一组反转链表

    之前的文章「递归反转链表的一部分」讲了如何递归地反转一部分链表,有读者就问如何迭代地反转链表,这篇文章解决的问题也需要反转链表的函数,我们不妨就用迭代方式来解决. 本文要解决「K 个一组反转链表」,不 ...

  10. java常用类——包装类

    八种基本数据类型对应八种包装类和它们的继承关系 基本数据类型 对应的包装类 boolean Boolean byte Byte short Short int Integer long Long ch ...