标签：Treeland LL back dfs maxn Choosing Capital include dp

https://codeforces.com/problemset/problem/219/D

思路:

n^2暴力dfs好写吧。

好写那就改成换根dp。

先dp[u]跑以u为根的子树结果。

再dfs一遍以u为根到整棵树的结果

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e5+1000;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar();	while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
vector<LL>g[maxn];
multiset<LL>s[maxn];
LL dp[maxn];
void predfs(LL u,LL fa){
     dp[u]=0;
     for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(v==fa) continue;
        predfs(v,u);
        if(s[u].find(v)==s[u].end()) dp[u]+=dp[v]+1;
        else dp[u]+=dp[v];
     }
}
void dfs(LL u,LL fa){

     for(LL i=0;i<g[u].size();i++){
        LL v=g[u][i];
        if(v==fa) continue;
        dp[v]=dp[u];
        if(s[u].find(v)!=s[u].end() ) dp[v]++;
        else dp[v]--;
        dfs(v,u);
     }
}
int main(void)
{
  cin.tie(0);std::ios::sync_with_stdio(false);
  LL n;cin>>n;
  for(LL i=1;i<n;i++){
    LL u,v;cin>>u>>v;
    g[u].push_back(v);
    g[v].push_back(u);
    s[u].insert(v);
  }
  predfs(1,0);
  dfs(1,0);
  LL mx=1e18;
  for(LL i=1;i<=n;i++){
    mx=min(mx,dp[i]);
  }
  vector<LL>v;
  for(LL i=1;i<=n;i++){
    if(dp[i]==mx){
        v.push_back(i);
    }
  }
  cout<<mx<<"\n";
  for(auto i:v){
    cout<<i<<" ";
  }
  cout<<"\n";
return 0;
}

标签：Treeland,LL,back,dfs,maxn,Choosing,Capital,include,dp
来源： https://blog.csdn.net/zstuyyyyccccbbbb/article/details/114701322

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。