質問:ツリーは、ルートノードから隣接ノードに兵士を送るか、他のノードの兵士に隣接ノードに移動するように指示するたびに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);
}
}