HDU 4358 Boring counting 莫隊演算法
阿新 • • 發佈:2019-02-19
題目大意:
就是現在給出一個有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; }