AcWing 347. 野餐规划 （控制度数的最小生成树，DFS）

（1）以Park作为1结点，题意即求1结点的度数不超过s的最小生成树（因为题目中说明每个人都可以开车，只要最后到park的车不超过 s即可）
（2）这样我们先做出除1结点之外的最小生成树森林，现在要考虑的就是用1结点把这些连通块连起来（如果s小于连通块数说明不可行，不过题目保证合法）
（3）贪心的想肯定是将每个连通块视为结点，加上1结点再做一个最小生成树，现在我们得到了一个答案的图
（4）这时候如果1的度数还有剩余，考虑加上不在答案图中的一条边（1，n，x），每加上这样一条边，我们就可以删除1----->x路径上最长的一条边，假设他是 (u，v，w)，这样答案就从原来的ans变成ans-w+x，我们一直进行这种加边删边的操作，直到不能再加（度数满了/无边可加）为止

int n;
int f[555];//并查集

int vis[555];
struct IN
{
    int x,y,z;//
}mx[555];//记录一个点到1点路径上最长的一条边的信息
vector<pair<int,pii> > edge;//用于最小生成树
vector<pii>edge1;//与1边相连的边
int road[555][555];//保存全部的路径信息 

int tag[555][555];//记录当前两点之间的边实际是否存在于生成的图中
int find(int x) 
{
    return x==f[x]?x:f[x]=find(f[x]);
}

void dfs(int x)
{
    vis[x]=1;
    for(int i=2;i<=n;++i)
    {
        if(tag[x][i]&&vis[i]==0)
        {
            if(road[x][i]<=mx[x].z) mx[i]=mx[x];
            else mx[i].x=x,mx[i].y=i,mx[i].z=road[x][i];
            dfs(i);
        }
    }
}
signed main()
{
    memset(road,0x3f,sizeof(road));
    int s,cnt=1;cin>>n;
    map<string,int>id;
    id["Park"]=1;
    while(n--)
    {
        string x,y;int u,v,w;cin>>x>>y>>w;
        if(!id[x]) id[x]=++cnt;
        if(!id[y]) id[y]=++cnt;
        u=id[x],v=id[y];
        edge.push_back(make_pair(w,make_pair(u,v)));
        road[u][v]=road[v][u]=min(road[u][v],w);
    }
    cin>>s;
    n=cnt;
    rpp(i,n) f[i]=i; 
    //做一次除1结点外的最小生成树
    sort(all(edge));
    ll ans=0;
    for(auto x:edge)
    {
        if(x.second.first==1||x.second.second==1) 
        {
            edge1.push_back(make_pair(x.first,x.second.first!=1?x.second.first:x.second.second));
            continue;
        }
        int fx=find(x.second.first),fy=find(x.second.second);
        if(fx!=fy) f[fx]=fy,ans+=x.first,tag[x.second.first][x.second.second]=tag[x.second.second][x.second.first]=1;//加到答案图中
    }
    //用1结点把生成的几棵树连起来
    sort(all(edge1));
    for(auto x:edge1)
    {
        int fx=1,fy=find(x.second);
        if(fx!=fy) f[fy]=fx,ans+=x.first,tag[1][x.second]=tag[x.second][1]=1,--s;
    }
    //还有多余的度数且可以优化的时候优化
    while(s--)
    {
        memset(vis,0,sizeof(vis));
        dfs(1);
        int flag=0,w=-1;//这里记得每次贪心的加一条收益最大的边
        for(auto x:edge1)
            if(tag[1][x.second]==0&&mx[x.second].z>x.first)
                if(mx[x.second].z-x.first>w) w=mx[x.second].z-x.first,flag=x.second;
        if(w!=-1) 
        {
            ans-=w;
            tag[mx[flag].x][mx[flag].y]=tag[mx[flag].y][mx[flag].x]=0;
            tag[1][flag]=tag[flag][1]=1;
        }
        else break;
    }
    cout<<"Total miles driven: "<<ans<<endl;
    return 0;
}