リンク:https://ac.nowcoder.com/acm/contest/904/E
出典:牛オフネットワーク
色のトンドン数
制限時間:C / C ++ 1秒、他の言語2秒
スペースの制約:C / C ++ 524288K、他の言語1048576K
64ビットIOフォーマット:%LLDの
対象の説明
トンドンは女の子の色数のようなものです、彼女は習得しています色のシーケンス番号の操作で、今、彼女は木の色の数は、今N点所定の方法を学ぶようになった、N-1個のツリーのエッジ(ルートによって店舗番号1)、各点は、Aは色は、常にそこにサブツリーをxはどのように多くの異なった色尋ねる、トンドンは、簡単にこの問題を解決するが、彼女はあなたがプログラミングされるクイズしようとしていました。
入力説明:
二つの整数Nの最初の行、M
各点の色を表すn個の整数の第2行
次に、N-1ラインの各U、Vは、(最終的なパターンが保証ツリーである)からV Uのエッジから双方向の存在を示します
2 <= N <= 100000,1 < = M、色<= N、
出力説明:
総m行:列出力対応する回答各問い合わせ
実施例1つ
の入力
コピー
。4. 3
1 3 1 2
1 2
2 3
1 4
1
2
4
出力
コピー
。3
2
1
アイデア:
私たちは木のDFSシーケンス内のノードのサブツリーのルートは、連続した範囲で、長さはサブツリーのサイズの範囲である知っているので、我々は時間DFSを維持することができます
iはノードのサイズをルートとするサブツリーに1
図2に示すように、i番目のノード順序をDFS。
3、DFSの配列におけるi番目のノードの位置。
その後、我々は、お問い合わせの問題の一般的な範囲に問題を置くことができ、我々は、Moのチームが行うオフラインアルゴリズムを使用することができます。
詳細コードを参照してください。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <iomanip>
#define ALL(x) (x).begin(), (x).end()
#define sz(a) int(a.size())
#define all(a) a.begin(), a.end()
#define rep(i,x,n) for(int i=x;i<n;i++)
#define repd(i,x,n) for(int i=x;i<=n;i++)
#define pii pair<int,int>
#define pll pair<long long ,long long>
#define gbtb ios::sync_with_stdio(false),cin.tie(0),cout.tie(0)
#define MS0(X) memset((X), 0, sizeof((X)))
#define MSC0(X) memset((X), '\0', sizeof((X)))
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define eps 1e-6
#define gg(x) getInt(&x)
#define chu(x) cout<<"["<<#x<<" "<<(x)<<"]"<<endl
using namespace std;
typedef long long ll;
ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;}
ll lcm(ll a, ll b) {return a / gcd(a, b) * b;}
ll powmod(ll a, ll b, ll MOD) {ll ans = 1; while (b) {if (b % 2)ans = ans * a % MOD; a = a * a % MOD; b /= 2;} return ans;}
inline void getInt(int* p);
const int maxn = 100010;
const int inf = 0x3f3f3f3f;
/*** TEMPLATE CODE * * STARTS HERE ***/
ll ans[maxn];
ll Ans=0ll;
int l=0;
int r=0;
struct node
{
int l,r,id;
}a[maxn];
int pos[maxn];
int n,m;
int len;
bool cmp(node aa,node bb)
{
if(pos[aa.l]==pos[bb.l])
{
return aa.r<bb.r;
}else
{
return pos[aa.l]<pos[bb.l];
}
}
int col[maxn];
int flag[maxn];
int wid[maxn];
void add(int x)
{
if(!flag[col[wid[x]]])
{
Ans++;
}
flag[col[wid[x]]]++;
}
void del(int x)
{
if(1==flag[col[wid[x]]])
{
Ans--;
}
flag[col[wid[x]]]--;
}
std::vector<int> son[maxn];
int S[maxn];
int tot=1;
int info[maxn];
void dfs(int x,int pre)
{
S[x]=1ll;
info[x]=tot;
wid[tot++]=x; // 保存dfs序的第i个位置是哪个节点。
for(auto y:son[x])
{
if(y!=pre)
{
dfs(y,x);
S[x]+=S[y];
}
}
}
int main()
{
//freopen("D:\\code\\text\\input.txt","r",stdin);
//freopen("D:\\code\\text\\output.txt","w",stdout);
gg(n);gg(m);
len=(int)(sqrt(n));
repd(i,1,n)
{
gg(col[i]);
}
int u,v;
repd(i,1,n-1)
{
gg(u);
gg(v);
son[v].pb(u);
son[u].push_back(v);
}
dfs(1,-1);
repd(i,1,m)
{
gg(a[i].l);
int temp=S[a[i].l];// 以a[i].l 为根的子树大小。
a[i].l=info[a[i].l]; // a[i].l 在dfs序中的位置。
a[i].r=a[i].l+temp-1; // 询问区间 右端点 在dfs序中的位置。
a[i].id=i;
pos[i]=i/len; // 莫队分块的部分。
}
sort(a+1,a+1+m,cmp);
repd(i,1,m)
{
while(l>a[i].l)
{
l--;
add(l);
}
while(r<a[i].r)
{
r++;
add(r);
}
while(l<a[i].l)
{
del(l);
l++;
}
while(r>a[i].r)
{
del(r);
r--;
}
ans[a[i].id]=Ans;
}
repd(i,1,m)
{
printf("%lld\n",ans[i]);
}
return 0;
}
inline void getInt(int* p) {
char ch;
do {
ch = getchar();
} while (ch == ' ' || ch == '\n');
if (ch == '-') {
*p = -(getchar() - '0');
while ((ch = getchar()) >= '0' && ch <= '9') {
*p = *p * 10 - ch + '0';
}
}
else {
*p = ch - '0';
while ((ch = getchar()) >= '0' && ch <= '9') {
*p = *p * 10 + ch - '0';
}
}
}