1. 程式人生 > >HDU 4358 Boring counting 莫隊演算法

HDU 4358 Boring counting 莫隊演算法

題目大意:

就是現在給出一個有N個結點的樹(N <= 100000)編號從1到N, 樹的根節點為1, 給定K, 每個點都有一個權值, 權值0 <= wi <= 10^9, 現在有Q次詢問, 對於每次詢問給出一個x代表詢問在編號x的結點及其子樹中, 出現恰好K次的取值有多少種?

大致思路:

第一次寫莫隊演算法.....

對於靜態的滿足能在O(1)的時間把[L, R]的答案轉移到[L, R + 1], [L - 1, R]的問題, 可以考慮莫隊演算法, 這題先將N個結點的後序遍歷順序處理出來, 那麼每次詢問就相當於在這個數列中詢問連續的一段中恰好出現K次的數的種數, 於是先將權值離散化然後就可以計數了

將Q次詢問離線出來分塊, 按照右端點遞增排序, 根據莫隊演算法, 複雜度在O(Nsqrt(N))級別

程式碼如下:

Result  :  Accepted     Memory  :  10928 KB     Time  :  1279 ms

/*
 * Author: Gatevin
 * Created Time:  2015/8/2 18:52:13
 * File Name: Sakura_Chiyo.cpp
 */
#pragma comment(linker, "/STACK:16777216") 
#include<iostream>
#include<sstream>
#include<fstream>
#include<vector>
#include<list>
#include<deque>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<bitset>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cctype>
#include<cmath>
#include<ctime>
#include<iomanip>
using namespace std;
const double eps(1e-8);
typedef long long lint;

#define maxn 100100

int N, K, Q;
vector<int> V;
int w[maxn];
int L[maxn], R[maxn];
int tim;
vector<int> G[maxn];
int ans;

struct Ask
{
    int l, r, pos, id;
    Ask(int _l, int _r, int _pos, int _id) : l(_l), r(_r), pos(_pos), id(_id){}
    Ask(){}
};

bool cmp(Ask a1, Ask a2)
{
    return a1.pos < a2.pos || (a1.pos == a2.pos && a1.r < a2.r);
}

Ask ask[maxn];

void dfs(int now, int fa)
{
    L[now] = ++tim;
    for(int i = 0, sz = G[now].size(); i < sz; i++) if(G[now][i] != fa)
        dfs(G[now][i], now);
    R[now] = tim;
    return;
}

int result[maxn];
int sum[maxn];

void add(int w)
{
    if(sum[w] == K) ans--;
    sum[w]++;
    if(sum[w] == K) ans++;
    return;
}

void del(int w)
{
    if(sum[w] == K) ans--;
    sum[w]--;
    if(sum[w] == K) ans++;
    return;
}

int main()
{
    int T;
    scanf("%d", &T);
    for(int cas = 1; cas <= T; cas++)
    {
        scanf("%d %d", &N, &K);
        V.clear();
        for(int i = 1; i <= N; i++)
        {
            scanf("%d", w + i);
            V.push_back(w[i]);
        }
        sort(V.begin(), V.end());
        for(int i = 1; i <= N; i++)
            w[i] = lower_bound(V.begin(), V.end(), w[i]) - V.begin();//離散化
        int u, v;
        for(int i = 1; i <= N; i++) G[i].clear();
        for(int i = 1; i < N; i++)
        {
            scanf("%d %d", &u, &v);
            G[u].push_back(v);
            G[v].push_back(u);
        }
        tim = 0;
        dfs(1, -1);//處理出時間戳, 實際上也就是樹的後續遍歷的順序
        scanf("%d", &Q);
        int x;
        int siz = sqrt(N*1.) + 4;
        for(int i = 0; i < Q; i++)
        {
            scanf("%d", &x);
            ask[i].l = L[x], ask[i].r = R[x];
            ask[i].pos = L[x] / siz, ask[i].id = i;
        }
        sort(ask, ask + Q, cmp);
        
        printf("Case #%d:\n", cas);
        
        memset(sum, 0, sizeof(sum));
        int l = 1, r = 0;
        ans = 0;
        add(w[++r]);
        for(int i = 0; i < Q; i++)
        {
            while(r < ask[i].r) add(w[++r]);
            while(r > ask[i].r) del(w[r--]);
            while(l < ask[i].l) del(w[l++]);
            while(l > ask[i].l) add(w[--l]);
            result[ask[i].id] = ans;
        }
        for(int i = 0; i < Q; i++)
            printf("%d\n", result[i]);
        if(cas != T) puts("");
    }
    return 0;
}