Codeforces375D Tree and Queries
题目链接
题目大意
给定一棵 \(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的更多相关文章
- [Codeforces375D]Tree and Queries(莫队算法)
题意:给定一棵树,每个节点有颜色,对于每个询问(u,k)询问以u为根节点的子树下有多少种颜色出现次数>=k 因为是子树,跟dfs序有关,转化为一段区间,可以用莫队算法求解 直接用一个数组统计出现 ...
- [Codeforces Round #221 (Div. 1)][D. Tree and Queries]
题目链接:375D - Tree and Queries 题目大意:给你一个有n个点的树,每个点都有其对应的颜色,给出m次询问(v,k),问v的子树中有多少种颜色至少出现k次 题解:先对所有的询问进行 ...
- Codeforces 375D Tree and Queries(DFS序+莫队+树状数组)
题目链接 Tree and Queries 题目大意 给出一棵树和每个节点的颜色.每次询问$vj, kj$ 你需要回答在以$vj$为根的子树中满足条件的的颜色数目, 条件:具有该颜色的节点数量至少 ...
- CodeForces 375D Tree and Queries 莫队||DFS序
Tree and Queries 题意:有一颗以1号节点为根的树,每一个节点有一个自己的颜色,求出节点v的子数上颜色出现次数>=k的颜色种类. 题解:使用莫队处理这个问题,将树转变成DFS序区间 ...
- codeforces 375D:Tree and Queries
Description You have a rooted tree consisting of n vertices. Each vertex of the tree has some color. ...
- CF375D Tree and Queries
题意翻译 给出一棵 n 个结点的树,每个结点有一个颜色 c i . 询问 q 次,每次询问以 v 结点为根的子树中,出现次数 ≥k 的颜色有多少种.树的根节点是1. 感谢@elijahqi 提供的翻译 ...
- 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 ...
- Codeforces 375 D Tree and Queries
Discription You have a rooted tree consisting of n vertices. Each vertex of the tree has some color. ...
- 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 ...
随机推荐
- Parcelable使用(一)
android有两种序列化方式:一是实现Serializable接口(是JavaSE本身就支持的),二是实现Parcelable接口(是Android特有功能,效率比实现Serializable接口高 ...
- eclipse时一直卡在进程中
(1)今天遇到进入eclipse时一直卡在 进程中,无论是重启电脑,还是重启软件 删除 D:\workspace\.metadata\.lock 文件才有用,特此记录下. (2)还有一种情况就是打开e ...
- D. Equalize the Remainders 解析(思維)
Codeforce 999 D. Equalize the Remainders 解析(思維) 今天我們來看看CF999D 題目連結 題目 略,請直接看原題 前言 感覺要搞個類似\(stack\)的東 ...
- Win32之创建进程
CreateProcess函数介绍 BOOL CreateProcessA( LPCSTR lpApplicationName, //可执行文件的名称完整的路径+程序名字) LPSTR lpComma ...
- 签到功能,用 MySQL 还是 Redis ?
现在的网站和app开发中,签到是一个很常见的功能,如微博签到送积分,签到排行榜. 如移动app ,签到送流量等活动. 用户签到是提高用户粘性的有效手段,用的好能事半功倍! 下面我们从技术方面看看常 ...
- linux下生成动态库和链接动态库
1.生成动态库 src/test.h #ifndef _TEST_H_HH #define _TEST_H_HH void print(); #endif src/test.cpp #include ...
- Centos7中MySQL8.0安装过程及遇到的问题记录
本文主要参考以下2篇文章,安装操作 https://www.miboxapp.com/article/detail/1146659339214393344 https://blog.csdn.net/ ...
- Django项目登录注册系统
Django项目之个人网站 关注公众号"轻松学编程"了解更多. Github地址:https://github.com/liangdongchang/MyWeb.git 感兴趣的可 ...
- 如何k个一组反转链表
之前的文章「递归反转链表的一部分」讲了如何递归地反转一部分链表,有读者就问如何迭代地反转链表,这篇文章解决的问题也需要反转链表的函数,我们不妨就用迭代方式来解决. 本文要解决「K 个一组反转链表」,不 ...
- java常用类——包装类
八种基本数据类型对应八种包装类和它们的继承关系 基本数据类型 对应的包装类 boolean Boolean byte Byte short Short int Integer long Long ch ...