zoukankan      html  css  js  c++  java
  • 【cf1009F】F. Dominant Indices(长链剖分)

    传送门

    长链剖分模板题。
    因为转移跟深度有关,所以我们直接继承深度最大的儿子的信息,其余暴力转移即可。
    继承不可能枚举,否则和直接转移复杂度无异,这里利用指针即可。
    其余暴力转移根据长链的性质复杂度不超过(O(n))
    代码如下:

    /*
     * Author:  heyuhhh
     * Created Time:  2020/6/9 9:43:02
     */
    #include <iostream>
    #include <algorithm>
    #include <cstring>
    #include <cstdio>
    #include <vector>
    #include <cmath>
    #include <set>
    #include <map>
    #include <queue>
    #include <iomanip>
    #include <assert.h>
    #include <functional>
    #include <numeric>
    #define MP make_pair
    #define fi first
    #define se second
    #define pb push_back
    #define sz(x) (int)(x).size()
    #define all(x) (x).begin(), (x).end()
    #define INF 0x3f3f3f3f
    #define Local
    #ifdef Local
      #define dbg(args...) do { cout << #args << " -> "; err(args); } while (0)
      void err() { std::cout << std::endl; }
      template<typename T, typename...Args>
      void err(T a, Args...args) { std::cout << a << ' '; err(args...); }
      template <template<typename...> class T, typename t, typename... A> 
      void err(const T <t> &arg, const A&... args) {
      for (auto &v : arg) std::cout << v << ' '; err(args...); }
    #else
      #define dbg(...)
    #endif
    using namespace std;
    typedef long long ll;
    typedef pair<int, int> pii;
    //head
    const int N = 1e6 + 5;
     
    int n;
    vector <int> G[N];
    int bson[N], len[N], ans[N];
    int *f[N], tmp[N], *id = tmp;
     
    void dfs(int u, int fa) {
        int Max = 0;
        for (auto v : G[u]) if (v != fa) {
            dfs(v, u);
            if (len[v] > Max) {
                Max = len[v];
                bson[u] = v;
            }
        }
        len[u] = len[bson[u]] + 1;
    }
     
    void dfs2(int u, int fa) {
        f[u][0] = 1;
        if (bson[u]) {
            f[bson[u]] = f[u] + 1;
            dfs2(bson[u], u);
            if (ans[u] != ans[bson[u]]) {
                ans[u] = ans[bson[u]] + 1;
            }
        }
        for (auto v : G[u]) {
            if (v == fa || v == bson[u]) continue;
            f[v] = id, id += len[v];
            dfs2(v, u);
            for (int i = 1; i <= len[v]; i++) {
                f[u][i] += f[v][i - 1];
                if (f[u][ans[u]] < f[u][i]) {
                    ans[u] = i;
                } else if (f[u][ans[u]] == f[u][i]) {
                    if (i < ans[u]) {
                        ans[u] = i;
                    }
                }
            }
        }
    }
     
    void run() {
        cin >> n;
        for (int i = 1; i < n; i++) {
            int u, v; cin >> u >> v;
            G[u].push_back(v);
            G[v].push_back(u);
        }
        dfs(1, 0);
        f[1] = id, id += len[1];
        dfs2(1, 0);
        for (int i = 1; i <= n; i++) {
            cout << ans[i] << '
    ';
        }
    }
     
    int main() {
        ios::sync_with_stdio(false);
        cin.tie(0); cout.tie(0);
        cout << fixed << setprecision(20);
        run();
        return 0;
    }
    
  • 相关阅读:
    前端-【学习心得】-事件委托方法
    [方法] iOS时间戳的转换
    [封装] 修改NSString中部分字段的颜色
    Python3基础16——file对象测试数据的读写与操作
    Python3基础15—— 变量作用域
    Python3基础14——函数&内置函数
    Python3基础13——冒泡排序
    Python3基础12——while循环
    Python3基础11——打印三角形
    Python3基础10——切片(str和list)
  • 原文地址:https://www.cnblogs.com/heyuhhh/p/13089335.html
Copyright © 2011-2022 走看看