cf1009F. Dominant Indices

題意：

1號節點為根，問對于以每個點為根的子樹種，求某個深度節點最多的層數。如果多個相等，輸出深度小的。

題解：

直接dsu就完事了，相當于是求子樹的眾數，但是注意常數，一開始因為常數過大在第100點wa了
將可以合并的dfs套在一起寫，減少常數
好像線段樹合并也可以做，但我還不會emm

代碼：

// Problem: F. Dominant Indices // Contest: Codeforces - Educational Codeforces Round 47 (Rated for Div. 2) // URL: https://codeforces.com/contest/1009/problem/F // Memory Limit: 512 MB // Time Limit: 4500 ms // Data:2021-09-03 11:43:05 // 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 } int n; const int maxn= 2e6 + 9; vector<int> vec[maxn]; int dep[maxn]; int siz[maxn]; int son[maxn]; int Son; void dfs(int u, int fa) {dep[u]= dep[fa] + 1;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 ans[maxn]; int num[maxn]; int Ans, Num; void cal(int u, int fa, int val) {num[dep[u]]+= val;if (val == 1) {if (num[dep[u]] > Num || (num[dep[u]] == Num && dep[u] < Ans)) {Ans= dep[u];Num= num[dep[u]];}}for (auto v : vec[u]) {if (v == fa || v == Son)continue;cal(v, u, val);} } // PII solve(int u, int fa, int rt) // { // PII maxx; // maxx.first= num[dep[u]]; // maxx.second= dep[u] - dep[rt]; // for (auto v : vec[u]) { // if (v == fa) // continue; // PII ans= solve(v, u, rt); // if (ans.first > maxx.first) { // maxx.first= ans.first; // maxx.second= ans.second; // } // if (ans.first == maxx.first) { // maxx.second= min(maxx.second, ans.second); // } // } // return maxx; // } 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);// PII p= solve(u, fa, u);ans[u]= Ans - dep[u];Son= 0;if (!keep) {cal(u, fa, -1);Num= 0;Ans= 0;} } int main() {//rd_test();read(n);for (int i= 1; i < n; i++) {int x, y;read(x, y);vec[x].push_back(y);vec[y].push_back(x);}dfs(1, 0);dfs2(1, 0, 0);for (int i= 1; i <= n; i++)printf("%d\n", ans[i]);//Time_test(); }

總結

以上是生活随笔為你收集整理的cf1009F. Dominant Indices的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。