距离和(换根dp)

简介: 距离和(换根dp)

传送门

/*********************************************************************
    程序名:
    版权: Joecai
    作者: Joecai
    日期: 2022-04-13 23:47
    说明:
*********************************************************************/
#include <bits/stdc++.h>
using namespace std;
#define x first
#define y second
# define rep(i,be,en) for(int i=be;i<=en;i++)
# define pre(i,be,en) for(int i=be;i>=en;i--)
#define ll long long
#define endl "\n"
#define LOCAL
#define pb push_back
#define int long long
typedef pair<ll, ll> PII;
#define eb emplace_back
#define sp(i) setprecision(i)
const int N = 2e5 + 10, INF = 0x3f3f3f3f;
ll qpow(int a, int b) {
  ll ans = 1;
  while (b) {
    if (b & 1) ans = ans * a;
    a = (ll)a * a;
    b >>= 1;
  }
  return ans;
}
inline int read()
{
  int X = 0;
  bool flag = 1;
  char ch = getchar();
  while (ch < '0' || ch > '9') {
    if (ch == '-') flag = 0;
    ch = getchar();
  }
  while (ch >= '0' && ch <= '9') {
    X = (X << 1) + (X << 3) + ch - '0';
    ch = getchar();
  }
  if (flag) return X;
  return ~(X - 1);
}
inline void write(int X)
{
  if (X < 0) {
    X = ~(X - 1);
    putchar('-');
  }
  if (X > 9) write(X / 10);
  putchar(X % 10 + '0');
}
struct dsu {
  private:
    vector<int> par, ran, cnt;
  public:
    dsu(int n) {
      par.resize(n, 0);
      iota(par.begin(), par.end(), 0);
      ran.resize(n, 0);
      cnt.resize(n, 1);
    }
    int find(int x) {
      if (par[x] == x)return x;
      else return par[x] = find(par[x]);
    }
    bool unite(int x, int y) {
      x = find(x), y = find(y);
      if (x == y) return false;
      if (ran[x] < ran[y]) {
        par[x] = y;
        cnt[y] += cnt[x];
      }
      else {
        par[y] = x;
        cnt[x] += cnt[y];
        if (ran[x] == ran[y])ran[x]++;
      }
      return true;
    }
    bool same(int x, int y) {
      return find(x) == find(y);
    }
    int getcnt(int x) {
      return cnt[find(x)];
    }
};
int n;
vector<int>g[N];
int sz[N];
int d[N];
int dfs1(int u, int f)
{
  sz[u] = 1;
  for (auto x : g[u])
  {
    if (x == f) continue;
    sz[u] += dfs1(x, u);
    d[u] += d[x] + sz[x];
  }
  return sz[u];
}
void dfs(int u, int f)
{
  for (auto x : g[u])
  {
    if (f == x) continue;
    d[x] = d[u] - 2 * sz[x] + n;
    dfs(x, u);
  }
}
void solve()
{
  cin >> n;
  for (int i = 1; i < n; i++)
  {
    int x, y;
    cin >> x >> y;
    g[x].push_back(y);
    g[y].push_back(x);
  }
  dfs1(1, -1); // 求1到每个带点的距离
  //cout << d[1] << endl;
  dfs(1, -1);
  for (int i = 1; i <= n; i++)
  {
    cout << d[i] << endl;
  }
}
signed main() {
  std::ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  //#ifdef LOCAL
  //freopen("data.in.txt","r",stdin);
  //freopen("data.out.txt","w",stdout);
  //#endif
  int __ = 1;
  //cin>>__;
  while (__--)
  {
    solve();
  }
  return 0;
}


目录
相关文章
|
3月前
|
算法 机器人 测试技术
【动态规划】【前缀和】【推荐】2463. 最小移动总距离
【动态规划】【前缀和】【推荐】2463. 最小移动总距离
|
4月前
|
消息中间件 Kubernetes JavaScript
动态规划-区间、计数类DP问题总结
动态规划-区间、计数类DP问题总结
|
7月前
|
算法
求最大连续子段和 的 dp算法
对于这样的问题,我们可以直接用暴力,一个双重循环,虽说可以,但也没有更高明的方法? 我们再分析这个问题,如果我们知道了某个数前面一段数的和,我们就该考虑把这个数加入到前一段,还是重新开始一段。这个地方很重要,如果前一段的和小于0,我们重新建一段,反之加到前一段。这样我们就可以把n个数分成几段了,且每一段都求出了他们的和,然后再循环一次求出最大的一个和,我们就得到想要的结果了,也可以在分段的时候直接求结果。
28 0
|
7月前
|
决策智能
【dp】背包问题
【dp】背包问题
323 2
|
8月前
|
机器学习/深度学习 人工智能
|
11天前
树形dp常见类型——换根dp
树形dp常见类型——换根dp
|
11天前
D. Book of Evil(树的直径+换根dp)
D. Book of Evil(树的直径+换根dp)
|
5月前
|
算法 程序员 索引
【算法训练-动态规划 四】【二维DP问题】最大正方形、最小路径和、不同路径
【算法训练-动态规划 四】【二维DP问题】最大正方形、最小路径和、不同路径
62 0
|
6月前
|
C++
9.DP单调队列优化
先弄出朴素DP->在用单调队列优化 一般都是区间的最大最小值,而且滑动的,才用单调队列优化
30 0
|
11月前
|
存储 算法
dp 就 dp ,数位dp是什么意思 ?
dp 就 dp ,数位dp是什么意思 ?
227 0