首页 > 其他分享 >CF375D Tree and Queries dsu on tree

CF375D Tree and Queries dsu on tree

时间:2022-10-03 09:56:17浏览次数:47  
标签:ch sum dsu Tree long CF375D MAXN include define

一颗以1为根树,每个点有颜色,每次询问求x子树内颜色出现次数>=k的数量。

线段树合并很难处理,考虑dfu on tree。

直接的想法是开一个数量树状数组 每次查一下1~k-1的前缀和来求解答案。

不过这并不够优秀,仔细思考一个颜色从1加到w这个过程是一次一次加的也就是说可以维护一个\(sum_i\)表示颜色数量>=i出现的次数。

这样修改查询复杂度都是线性的。

同时可以发现莫队也可以做这个子树内的问题。

code
//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<ctime>
#include<cctype>
#include<queue>
#include<deque>
#include<stack>
#include<iostream>
#include<iomanip>
#include<cstdio>
#include<cstring>
#include<string>
#include<ctime>
#include<cmath>
#include<cctype>
#include<cstdlib>
#include<queue>
#include<deque>
#include<stack>
#include<vector>
#include<algorithm>
#include<utility>
#include<bitset>
#include<set>
#include<map>
#define ll long long
#define db double
#define INF 1000000000
#define inf 100000000000000000ll
#define ldb long double
#define pb push_back
#define put_(x) printf("%d ",x);
#define get(x) x=read()
#define gt(x) scanf("%d",&x)
#define gi(x) scanf("%lf",&x)
#define put(x) printf("%d\n",x)
#define putl(x) printf("%lld\n",x)
#define rep(p,n,i) for(int i=p;i<=n;++i)
#define go(x) for(int i=lin[x],tn=ver[i];i;tn=ver[i=nex[i]])
#define pii pair<int,int>
#define mk make_pair
#define P 1000000007ll
#define gf(x) scanf("%lf",&x)
#define pf(x) ((x)*(x))
#define uint unsigned long long
#define ui unsigned
#define EPS 1e-10
#define sq sqrt
#define a(x) t[x].a
#define sum(x) t[x].sum
#define b(x) t[x].b
#define F first
#define S second
using namespace std;
char *fs,*ft,buf[1<<15];
inline char gc()
{
    return (fs==ft&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),fs==ft))?0:*fs++;
}
inline int read()
{
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=gc();}
    return x*f;
}
const int MAXN=100010;
int n,Q,len;
vector<int>q[MAXN];
int c[MAXN],sum[MAXN],ans[MAXN],v[MAXN];
int b[MAXN],lin[MAXN],nex[MAXN<<1],ver[MAXN<<1],son[MAXN],sz[MAXN];
inline void add(int x,int y)
{
	ver[++len]=y;nex[len]=lin[x];lin[x]=len;
	ver[++len]=x;nex[len]=lin[y];lin[y]=len;
}
inline void dfs(int x,int fa)
{
	sz[x]=1;
	for(int i=lin[x];i;i=nex[i])
	{
		int tn=ver[i];
		if(tn==fa)continue;
		dfs(tn,x);
		sz[x]+=sz[tn];
		if(sz[son[x]]<sz[tn])son[x]=tn;
	}
}
inline void get_ans(int x,int fa,int w,int tr)
{
	if(w==-1){--sum[c[v[x]]];--c[v[x]];}
	else {++c[v[x]];++sum[c[v[x]]];}
	for(int i=lin[x];i;i=nex[i])
	{
		int tn=ver[i];
		if(tn==fa||tn==tr)continue;
		get_ans(tn,x,w,tr);
	}
}
inline void dsu(int x,int fa)
{
	for(int i=lin[x];i;i=nex[i])
	{
		int tn=ver[i];
		if(tn==fa||tn==son[x])continue;
		dsu(tn,x);
		get_ans(tn,x,-1,0);
	}
	if(son[x])dsu(son[x],x);
	get_ans(x,fa,1,son[x]);
	for(int i=0;i<(int)q[x].size();++i)
	{
		int ww=q[x][i];
		ans[ww]=sum[b[ww]];
	}
}
signed main()
{
	freopen("1.in","r",stdin);
	n=read();Q=read();
	rep(1,n,i)v[i]=read();
	rep(2,n,i)add(read(),read());
	rep(1,Q,i)
	{
		int x=read();b[i]=read();
		q[x].push_back(i);
	}
	dfs(1,0);
	dsu(1,0);
	rep(1,Q,i)printf("%d\n",ans[i]);
	return 0;
}

标签:ch,sum,dsu,Tree,long,CF375D,MAXN,include,define
From: https://www.cnblogs.com/chdy/p/16750043.html

相关文章

  • CF1624G MinOr Tree 题解
    CF1624GMinOrTree给定\(n\)个点,\(m\)条边,求在或运算小的最小生成树考虑二进制拆位,从高位玩往地位贪心,如果答案第\(i\)位可以为\(0\),后\(i-1\)取值无论是多少......
  • leetcode -- tree 4
    不同的二叉搜索树II递归#Definitionforabinarytreenode.#classTreeNode:#def__init__(self,val=0,left=None,right=None):#self.val=......
  • [ 数据结构 - C++]红黑树RBTree
    在上篇文章我们了解了第一种平衡二叉搜索树AVL树,我们知道AVL树是通过平衡因子来控制左右子树高度差,从而将二叉树变成一颗平衡二叉搜索树。本篇文章我们将要了解另外一种平衡......
  • New Year Tree
    NewYearTreeTranslation给出一棵\(n\)个节点的树,根节点为\(1\)。每个节点上有一种颜色\(c_i\)。\(m\)次操作。操作有两种:1uc:将以\(u\)为根的子树上的所有节点......
  • leetcode 538. Convert BST to Greater Tree 把二叉搜索树转换为累加树(简单)
    一、题目大意给出二叉搜索树的根节点,该树的节点值各不相同,请你将其转换为累加树(GreaterSumTree),使每个节点node的新值等于原树中大于或等于node.val的值之和。提......
  • npm卸载"Tracker idealTree already exists"
    问题使用npm卸载babel插件的时候执行命令npmuninstallbabel...出现如下报错npmERR!Tracker"idealTree"alreadyexists在执行卸载命令的时候是在根目录下执行的,......
  • leetcode 513. Find Bottom Left Tree Value 找树左下角的值 (简单)
    一、题目大意给定一个二叉树的根节点root,请找出该二叉树的最底层最左边节点的值。假设二叉树中至少有一个节点。示例1:输入:root=[2,1,3]输出:1示例2:......
  • leetcode -- tree 3
    使用归并排序简单解决问题归并排序用传统方法归并classSolution:defsortArray(self,nums:List[int])->List[int]:defmergesort(nums:List[int......
  • Game on Tree 3
    ProblemStatementThereisarootedtreewith$N$vertices,Vertex$1$beingtheroot.Foreach$i=1,2,\ldots,N-1$,the$i$-thedgeconnectsVertex$u_i$......
  • TreeView的RenderControl的问题
    TreeView,这东西,正常情况下一般是不用的,不过我们的美工,没弄个树型的样式出来,没折,将就用一下TreeView了说重点:环境搭建:一页面,拖一下TreeView控件上去,随便添加几个项。然后Page......