zoukankan      html  css  js  c++  java
  • 【CF375D】Tree and Queries

    题目

    题目链接:https://codeforces.com/problemset/problem/375/D
    给定一棵 (n) 个节点的有根树,节点有颜色,(m) 次询问一个点的子树内出现次数 (geq k) 的颜色数有多少。
    (nleq 10^5)

    思路

    dsu on tree 板子题。
    本来太板子不想写的,但是因为太久没写了就水了一道题。
    直接维护一个桶表示出现次数等于下标的颜色数量即可。
    时间复杂度 (O(nlog n+m))

    代码

    #include <bits/stdc++.h>
    #define mp make_pair
    #define fi first
    #define se second
    using namespace std;
    
    const int N=100010;
    int n,m,tot,head[N],a[N],cnt[N],sum[N],ans[N],siz[N],son[N],id[N],rk[N];
    vector<pair<int,int> > ask[N];
    
    struct edge
    {
    	int next,to;
    }e[N*2];
    
    void add(int from,int to)
    {
    	e[++tot]=(edge){head[from],to};
    	head[from]=tot;
    }
    
    void dfs1(int x,int fa)
    {
    	siz[x]=1;
    	for (int i=head[x];~i;i=e[i].next)
    	{
    		int v=e[i].to;
    		if (v!=fa)
    		{
    			dfs1(v,x); siz[x]+=siz[v];
    			if (siz[v]>siz[son[x]]) son[x]=v;
    		}
    	}
    }
    
    void dfs2(int x,int fa,bool flag)
    {
    	id[x]=++tot; rk[tot]=x;
    	for (int i=head[x];~i;i=e[i].next)
    	{
    		int v=e[i].to;
    		if (v!=fa && v!=son[x]) dfs2(v,x,1);
    	}
    	if (son[x]) dfs2(son[x],x,0);
    	cnt[a[x]]++; sum[cnt[a[x]]]++;
    	if (son[x])
    		for (int i=id[x]+1;i<id[son[x]];i++)
    			cnt[a[rk[i]]]++,sum[cnt[a[rk[i]]]]++;
    	for (int i=0;i<ask[x].size();i++)
    		ans[ask[x][i].se]=sum[ask[x][i].fi];
    	if (flag)
    		for (int i=id[x];i<id[x]+siz[x];i++)
    			sum[cnt[a[rk[i]]]]--,cnt[a[rk[i]]]--;
    }
    
    int main()
    {
    	memset(head,-1,sizeof(head));
    	scanf("%d%d",&n,&m);
    	for (int i=1;i<=n;i++)
    		scanf("%d",&a[i]);
    	for (int i=1,x,y;i<n;i++)
    	{
    		scanf("%d%d",&x,&y);
    		add(x,y); add(y,x);
    	}
    	for (int i=1,x,y;i<=m;i++)
    	{
    		scanf("%d%d",&x,&y);
    		ask[x].push_back(mp(y,i));
    	}
    	tot=0; 
    	dfs1(1,0); dfs2(1,0,1);
    	for (int i=1;i<=m;i++) cout<<ans[i]<<"
    ";
    	return 0;
    }
    
  • 相关阅读:
    生成doc
    Struts2 配置
    Struts(一)struts2.2概述与MVC模式深度剖析
    [DOM Event Learning] Section 4 事件分发和DOM事件流
    jquery 高亮显示案例
    jquery 下拉菜单
    3_1:操作样式
    2_2:筛选器
    2_1:选择元素
    本地存储小案例:记住用户名
  • 原文地址:https://www.cnblogs.com/stoorz/p/14885357.html
Copyright © 2011-2022 走看看