upc2021个人训练赛第23场M: 紫罗兰(dsu)
问题 M: 紫罗兰
时间限制: 2 Sec 内存限制: 256 MB
题目描述
梦不知何时醒、何时灭,
纵然天崩地裂,也见不得天日,
原来都是青天白日下不敢细想的思量……
那是从来无处表白的,
那些生不得、死不得、忘不得也记不得的心。
流年那样无理残忍,稍有踟蹰,它就偷梁换柱,叫人撕心裂肺,再难回头。
在附中的小路旁种着许多紫罗兰,而我们的问题正是与这些紫罗兰有关。
我们可以把附中里的道路系统抽象成一棵树,而所有的紫罗兰都被种植在树的节点上,并且每一个节点上的紫罗兰都是同一种品种。我们用一个数字来表示一种品种。同时,我们给每一个节点一个编号,并规定 111 号节点为根节点。每次,Marser 想要知道,以编号为 iii 的节点为根的子树中,出现次数最多,且在此情况下品种编号最小的品种。即意,求出该子树中点权的最小众数。现在,您被要求解决这一问题。
输入
第一行一个整数 n,表示树的节点的数量。
接下来一行 n 个整数 ai,表示每个节点上的品种编号。
接下来 n−1 行,每行两个整数 s,t,描述树上的一条边。
接下来一行一个整数 q,表示 Marser 的询问的数量。
接下来 q 行,每行一个整数 x,表示询问以节点 x 为根的子树中的最小众数。
输出
对于每个询问,输出相应的答案。
样例输入 Copy
5
3 2 3 1 3
1 2
1 3
3 4
3 5
3
1
3
4
样例输出 Copy
3
3
1
提示
对于100%的数据,保证 n,q≤106 ,1≤ai≤109
本题输入文件较大,请注意使用读入优化,并采用复杂度较为优秀的做法。
思路:
讲解链接
\(dsu\)模板题,每次更新答案时都取最小的众数。复杂度\(O(nlogn)\)
代码:
// Problem: CF600E Lomsat gelral
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/CF600E
// Memory Limit: 250 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>PLL;
typedef pair<int, int>PII;
typedef pair<double, double>PDD;
#define I_int ll
inline ll read()
{
ll x = 0, f = 1;
char ch = getchar();
while(ch < '0' || ch > '9')
{
if(ch == '-')f = -1;
ch = getchar();
}
while(ch >= '0' && ch <= '9')
{
x = x * 10 + ch - '0';
ch = getchar();
}
return x * f;
}
inline void out(ll x){
if (x < 0) x = ~x + 1, putchar('-');
if (x > 9) out(x / 10);
putchar(x % 10 + '0');
}
inline void write(ll x){
if (x < 0) x = ~x + 1, putchar('-');
if (x > 9) write(x / 10);
putchar(x % 10 + '0');
puts("");
}
#define read read()
#define closeSync ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define multiCase int T;cin>>T;for(int t=1;t<=T;t++)
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define perr(i,a,b) for(int i=(a);i>(b);i--)
ll ksm(ll a, ll b, ll p)
{
ll res = 1;
while(b)
{
if(b & 1)res = res * a % p;
a = a * a % p;
b >>= 1;
}
return res;
}
const int inf = 0x3f3f3f3f;
#define PI acos(-1)
const int maxn=1000000+100;
int n,col[maxn];
vector<int>g[maxn];
ll ans[maxn],sum;
int son[maxn],siz[maxn],cnt[maxn],maxx,Son;
void dfs1(int u,int fa){
siz[u]=1;
for(int i=0;i<g[u].size();i++){
int j=g[u][i];
if(j==fa) continue;
dfs1(j,u);
siz[u]+=siz[j];
if(siz[j]>siz[son[u]]) son[u]=j;
}
}
void add(int u,int fa,int val){
cnt[col[u]]+=val;
if(cnt[col[u]]>maxx) maxx=cnt[col[u]],sum=col[u];
else if(cnt[col[u]]==maxx) sum=min(sum,col[u]*1ll);
for(int i=0;i<g[u].size();i++){
int j=g[u][i];
if(j==fa||j==Son) continue;
add(j,u,val);
}
}
void dfs2(int u,int fa,int op){
for(int i=0;i<g[u].size();i++){
int j=g[u][i];
if(j==fa) continue;
if(j!=son[u]) dfs2(j,u,0);
}
if(son[u]) dfs2(son[u],u,1),Son=son[u];
add(u,fa,1);Son=0;
ans[u]=sum;
if(!op) add(u,fa,-1),sum=0,maxx=0;
}
int main(){
n=read;
rep(i,1,n) col[i]=read;
rep(i,1,n-1){
int u=read,v=read;
g[u].push_back(v);
g[v].push_back(u);
}
dfs1(1,0);
dfs2(1,0,0);
int _=read;
while(_--){
int x=read;
printf("%lld\n",ans[x]);
}
return 0;
}