cf375D. Tree and Queries

題意：

給你一顆有根樹(shù)，每個(gè)點(diǎn)都有一個(gè)顏色，有m次詢(xún)問(wèn)，問(wèn)以u(píng)為根的子樹(shù)中，相同顏色數(shù)量超過(guò)k的有多少種顏色？

題解：

這個(gè)題做法很多，有莫隊(duì)分塊，這里用的dsu
就是對(duì)于一個(gè)子樹(shù)內(nèi)的顏色進(jìn)行記錄，為了方便我們后面的數(shù)量統(tǒng)計(jì)，對(duì)于一個(gè)子樹(shù)中所有顏色出現(xiàn)次數(shù)，我們用樹(shù)狀數(shù)組來(lái)維護(hù)。比如顏色1已經(jīng)出現(xiàn)了2兩次，又出現(xiàn)了一次，那么我們就在樹(shù)狀數(shù)組中將2的位置減1，將位置3加1，樹(shù)狀數(shù)組維護(hù)的是顏色出現(xiàn)的數(shù)量。這樣對(duì)于出現(xiàn)次數(shù)大于等于k的顏色種類(lèi)，我們直接query(n)-query(l-1)即可，注意k有可能回比n大，所以要注意特判。
還有樹(shù)狀數(shù)組的維護(hù)盡可能在dfs過(guò)程中，不要單獨(dú)再循環(huán)維護(hù)，不然常數(shù)太大

代碼：

// Problem: D. Tree and Queries // Contest: Codeforces - Codeforces Round #221 (Div. 1) // URL: https://codeforces.com/contest/375/problem/D // Memory Limit: 256 MB // Time Limit: 1000 ms // Data:2021-09-03 12:55:52 // By Jozky#include <bits/stdc++.h> #include <unordered_map> #define debug(a, b) printf("%s = %d\n", a, b); using namespace std; typedef long long ll; typedef unsigned long long ull; typedef pair<int, int> PII; clock_t startTime, endTime; //Fe~Jozky const ll INF_ll= 1e18; const int INF_int= 0x3f3f3f3f; void read(){}; template <typename _Tp, typename... _Tps> void read(_Tp& x, _Tps&... Ar) {x= 0;char c= getchar();bool flag= 0;while (c < '0' || c > '9')flag|= (c == '-'), c= getchar();while (c >= '0' && c <= '9')x= (x << 3) + (x << 1) + (c ^ 48), c= getchar();if (flag)x= -x;read(Ar...); } template <typename T> inline void write(T x) {if (x < 0) {x= ~(x - 1);putchar('-');}if (x > 9)write(x / 10);putchar(x % 10 + '0'); } void rd_test() { #ifdef LOCALstartTime= clock();freopen("in.txt", "r", stdin); #endif } void Time_test() { #ifdef LOCALendTime= clock();printf("\nRun Time:%lfs\n", (double)(endTime - startTime) / CLOCKS_PER_SEC); #endif } const int maxn= 2e5 + 9; int col[maxn]; vector<int> vec[maxn]; int siz[maxn]; int son[maxn], Son; int ans[maxn]; void dfs(int u, int fa) {siz[u]= 1;for (auto v : vec[u]) {if (v == fa)continue;dfs(v, u);siz[u]+= siz[v];if (siz[v] > siz[son[u]])son[u]= v;} } int va[maxn];vector<PII> q[maxn]; int f[maxn]; int num[maxn]; int n, m; int lowbits(int x) {return x & (-x); } void update(int pos, int val) {for (int i= pos; i <= n; i+= lowbits(i)) {va[i]+= val;} } int query(int pos) {int sum= 0;for (int i= pos; i; i-= lowbits(i)) {sum+= va[i];}return sum; } int query(int l, int r) {if (l > r)return 0;return query(r) - query(l - 1); } void cal(int u, int fa, int val) {if (num[col[u]] != 0)update(num[col[u]], -1);num[col[u]]+= val;if (num[col[u]] != 0)update(num[col[u]], 1);for (auto v : vec[u]) {if (v == fa || v == Son)continue;cal(v, u, val);} } void dfs2(int u, int fa, int keep) {for (auto v : vec[u]) {if (v == fa || v == son[u])continue;dfs2(v, u, 0);}if (son[u]) {dfs2(son[u], u, 1);Son= son[u];}cal(u, fa, 1);for (auto it : q[u]) {int k= it.first;int id= it.second;ans[id]= query(k, n);}Son= 0;if (!keep) {cal(u, fa, -1);} } int main() {//rd_test();read(n, m);for (int i= 1; i <= n; i++) {read(col[i]);}for (int i= 1; i < n; i++) {int u, v;read(u, v);vec[u].push_back(v);vec[v].push_back(u);}dfs(1, 0);for (int i= 1; i <= m; i++) {int v, k;read(v, k);q[v].push_back({k, i});// read(q[i].v,q[i].k);}dfs2(1, 0, 0);for (int i= 1; i <= m; i++)printf("%d\n", ans[i]);//Time_test(); } 創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯，堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)

總結(jié)

以上是生活随笔為你收集整理的cf375D. Tree and Queries的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。