Codeforces - 375D （ dfs序 + 莫隊 ）

Codeforces - 375D （ 莫隊 ）

題目鏈接:https://codeforces.com/contest/375/problem/D

題意：給出一棵 n 個結點的樹，每個結點有一個顏色 c i 。 詢問 q 次，每次詢問以 v 結點爲根的子樹中，出現次數 ≥k 的顏色有多少種。樹的根節點是1。

思路：dfs序上暴力莫隊。注意dfs序和原節點編號的轉換，用dfn和rev來保存。

具體實現的時候可以直接用 via[i] 表示第 i 個顏色的出現次數，ci [ i ] 表示出現次數多於 i 的顏色的種類

由於左右端點移動的時候只會對一個ci [ i ] 產生影響，所以修改是O(1)的

input

8 5
1 2 2 3 3 2 3 3
1 2
1 5
2 3
2 4
5 6
5 7
5 8
1 2
1 3
1 4
2 3
5 3

output

2
2
1
0
1

代碼：

#include <bits/stdc++.h>

using namespace std;

const int maxn = 2e5+10;
int dfn[maxn],col[maxn],via[maxn],sz[maxn];
int ans[maxn],ci[maxn],rev[maxn];
vector<int> G[maxn];
int n,m,tim,block;
struct node {
    int l,r,k,id;
}q[maxn];

int rule( node a, node b )
{
    if ( a.l/block==b.l/block ) return a.r<b.r;
    return  a.l/block<b.l/block;
}

void dfs( int u, int f )
{
    dfn[u] = ++tim;
    rev[tim] = u;
    sz[u] = 1;
    for ( int i=0; i<G[u].size(); i++ ) {
        int v = G[u][i];
        if ( v==f ) continue ;
        dfs(v,u);
        sz[u] += sz[v];
    }
}

void add( int x )
{
    x = rev[x];
    x = col[x];
    via[x] ++;
    ci[ via[x] ] ++;
}

void del( int x )
{
    x = rev[x];
    x = col[x];
    ci[ via[x] ] --;
    via[x] --;
}

int main()
{
    cin >> n >> m;
    for ( int i=1; i<=n; i++ ) scanf("%d",&col[i]);
    for ( int i=1; i<n; i++ ) {
        int u,v;scanf("%d %d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    tim = 0;
    dfs(1,0);
    for ( int i=0; i<m; i++ ) {
        int x;scanf("%d %d",&x,&q[i].k);
        q[i].l = dfn[x];q[i].id = i+1;
        q[i].r = dfn[x] + sz[x] - 1;
    }
    block = sqrt(m);
    sort(q,q+m,rule);
    int l = 1, r = 0;
    for ( int i=0; i<m; i++ ) {
        while ( q[i].l>l ) {
            del(l++);
        }
        while ( q[i].l<l ) {
            add(--l);
        }
        while ( q[i].r>r ) {
            add(++r);
        }
        while ( q[i].r<r ) {
            del(r--);
        }
        ans[ q[i].id ] = ci[ q[i].k ];
    }
    for ( int i=1; i<=m; i++ ) cout << ans[i] << endl;

    return 0;
}