ccpc2020 Qinhuangdao-Kingdom's Power(tree dp)

質問:ツリーは、ルートノードから隣接ノードに兵士を送るか、他のノードの兵士に隣接ノードに移動するように指示するたびに1回費やすことができます。兵士がツリー全体を横断するための最小コストを見つけます。1はルートノードは、最初はルートノードだけが兵士の
問題の解決策を持っています。
問題はすべてのリーフノードを完了するのに必要な時間に変換できます。
ルートノードとリーフノードの間の距離、および隣接するリーフノードとリーフノードの間の距離の最小値を維持する必要があります。最初に、iノードをルートノードとして、サブツリー内のリーフノードからiまでの最も遠い距離を計算します。次に、最も遠い距離を並べ替え、ツリーを再度トラバースして、サブツリー内のルートノードからiノードまでの距離と、iノードからリーフノードまでの最小距離を計算します。2つの最小値はiの最小値です。
コード:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<stack>
#include<set>
#define iss ios::sync_with_stdio(false)
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
typedef pair<int ,int > pii;
const ll mod=998244353;
const ll INF=0x3f3f3f3f;
const int MAXN=1e6+5;
int deep[MAXN];
int ans[MAXN];
std::vector<pii> v1[MAXN];
int dfs(int u,int f,int d)
{
    
    
    deep[u]=d;
    if(v1[u].size()==0) return 1;
    for(int i=0;i<v1[u].size();i++)
    {
    
    
        int v=v1[u][i].second;
        if(v==f) continue;
        int maxx=dfs(v,u,d+1);
        //cout<<v<<" "<<maxx<<endl;
        v1[u][i].first=maxx;
    }
    sort(v1[u].begin(),v1[u].end());
    return v1[u].back().first+1;

}
int solve(int u,int f,int d)
{
    
    
    ans[u]=d;
    if(v1[u].size()==0)
    {
    
    
        cout<<u<<" "<<d<<endl;
    }
    int minn=d;
    for(int i=0;i<v1[u].size();i++)
    {
    
    
        int v=v1[u][i].second;
        if(v==f) continue;
        minn=min(deep[u],solve(v,u,minn+1));
    }
    return minn+1;
}
int main()
{
    
    
    int t;
    int kase=1;
    scanf("%d",&t);
    while(t--)
    {
    
    
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
    
    
            v1[i].clear();
        }
        int fa;
        for(int i=2;i<=n;i++)
        {
    
    
            scanf("%d",&fa);
            v1[fa].push_back(make_pair(0,i));
        }
        dfs(1,-1,0);
        solve(1,-1,0);
        ll sum=0;
        for(int i=1;i<=n;i++)
        {
    
    
            if(v1[i].size()==0)
            {
    
    
                sum+=ans[i];
            }
        }
        printf("Case #%d: %lld\n",kase++,sum);
    }
}


おすすめ

転載: blog.csdn.net/weixin_45755679/article/details/109209514