比赛 2024暑假C班集训C 评测结果 TTTTTTTTTT
题目名称 喵星人集会 最终得分 0
用户昵称 djyqjy 运行时间 19.979 s
代码语言 C++ 内存使用 3.25 MiB
提交时间 2024-07-12 11:56:30
显示代码纯文本
#include<bits/stdc++.h>
using namespace std;
const int N=55;
int n;
string s;
int g[N][N];
int depth[N];
vector <int> r;
void dfs(int x,int fa)
{
    for(int i=1;i<=n;i++)
    {
        if(g[x][i]&&i!=fa)
        {
            depth[i]=depth[x]+1;
            dfs(i,x);
        }
    }
    return;
}
int ans;
int anss=INT_MAX;
int main()
{
    freopen("party.in","r",stdin);
    freopen("party.out","w",stdout);
    scanf("%d",&n);
    cin>>s;
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        g[u][v]=g[v][u]=1;
    }
    for(int i=0;i<n;i++)
    {
        if(s[i]) r.push_back(i+1);
    }
    for(int i=0;i<r.size();i++)
    {
        depth[r[i]]=1;
        dfs(r[i]+1,0);
        ans=depth[r[0]]+depth[r[1]]+depth[r[2]]-3*depth[r[i]]-2;
        int a=r[0],b=r[1];
        if(i==0) a=r[2];
        if(i==1) b=r[2];
        while(depth[a]!=2)
        {
            for(int j=1;j<=n;j++)
            {
                if(g[i][j]&&depth[j]<depth[a])
                {
                    a=j;
                    break;
                }
            }
        }
        while(depth[b]!=2)
        {
            for(int j=1;j<=n;j++)
            {
                if(g[i][j]&&depth[j]<depth[b])
                {
                    b=j;
                    break;
                }
            }
        }
        if(a==b) ans--;
        anss=min(anss,ans);
    }
    printf("%d",anss);
    return 0;
}