题意:
求满足一个点限定度数情况下的最小生成树。
题解:
这题写的是真的麻烦。不考虑限制度数的点park,将剩下几个连通块求出最小生成树,prime一边染色一边最小生成树即可,过程中使用的边进行标记。将park先连上各个连通块最短的边,此时的生成树并不是最优的。此时只有park剩下的边可能对减小结果,因此判断park剩下的边能不能减小贡献就行。对一个树,加上一个边一定会形成环。BFS出这个环,找到最大的边,减去加上的边就是贡献。跑到park的度满了或者加边没有贡献为止。
AC代码:
#include <cstdio>
#include <iostream>
#include <vector>
#include <string>
#include <queue>
#include <algorithm>
#include <cmath>
#include <set>
#include <map>
#include <iomanip>
#include <cstdlib>
#include <stack>
#include <cstring>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define repp(i,a,b) for(int i=(a);i<(b);i++)
#define lep(i,a,b) for(int i=(a);i>=(b);i--)
#define lepp(i,a,b) for(int i=(a);i>(b);i--)
#define pii pair<int,int>
#define pll pair<long long,long long>
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define All(x) x.begin(),x.end()
#define ms(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define INFF 0x3f3f3f3f3f3f3f3f
#define multi int T;scanf("%d",&T);while(T--)
using namespace std;
typedef long long ll;
typedef double db;
const int N=20+5;
const int mod=1e9+7;
const db eps=1e-6;
const db pi=acos(-1.0);
int n,park,c[N],s,cnt,dis[N];
map<string,int>num;
int road[N][N],used[N][N],to[N];
struct edge{
int u,v,w;
bool operator < (const edge &A) const{
return w<A.w;
}
};
int bfs(int x,int &u,int &v){
int pre[N];
ms(pre,0);
pre[x]=park;
queue<int>q;
q.push(x);
while(q.size()){
int cur=q.front();
q.pop();
if(cur==park) break;
rep(i,1,cnt)
if(used[cur][i]==1&&pre[i]==0){
pre[i]=cur;
q.push(i);
}
}
int w=0,cur=park;
while(pre[cur]!=park){
if(w<road[cur][pre[cur]]){
w=road[cur][pre[cur]];
u=cur,v=pre[cur];
}
cur=pre[cur];
}
return w;
}
int main(){
#ifndef ONLINE_JUDGE
freopen("D:\\work\\data.in","r",stdin);
#endif
int minn[N],n,bef[N];
cin>>n;
cnt=0;
ms(road,INF);
ms(dis,INF);
ms(bef,0);
rep(i,1,n){
string s1,s2;
int w;
cin>>s1>>s2>>w;
if(num.find(s1)==num.end()) num[s1]=++cnt;
if(num.find(s2)==num.end()) num[s2]=++cnt;
road[num[s1]][num[s2]]=road[num[s2]][num[s1]]=w;
if(s1=="Park") park=num[s1];
if(s2=="Park") park=num[s2];
}
int color=0,ans=0;
c[park]=++color;
rep(i,1,cnt){
if(c[i]) continue;
++color;
priority_queue<pii>q;
q.push(mp(0,i));
dis[i]=0;
while(q.size()){
int idx=q.top().second,w=-q.top().first;
q.pop();
if(c[idx]) continue;
c[idx]=color;
if(bef[idx]){
used[bef[idx]][idx]=1;
used[idx][bef[idx]]=1;
}
ans+=w;
// q.pop();pop要在push之前,要不然对于最小生成树这种并不是单调的,可能pop掉刚push的
rep(j,1,cnt){
if(c[j]) continue;
if(dis[j]>road[idx][j]){
dis[j]=road[idx][j];
bef[j]=idx;
q.push({-dis[j],j});
}
}
}
}
ms(minn,INF);
minn[c[park]]=0;
rep(i,1,cnt){
if(minn[c[i]]>road[park][i]){
minn[c[i]]=road[park][i];
to[c[i]]=i;
}
}
rep(i,2,color){
ans+=minn[i];
used[park][to[i]]=used[to[i]][park]=1;
}
cin>>s;
rep(i,1,s-color+1){
edge e,maxn;
int pos;
maxn.w=0;
rep(j,1,cnt){
if(used[park][j]==1||road[park][j]==INF) continue;
e.w=bfs(j,e.u,e.v)-road[park][j];
if(e.w>maxn.w){maxn=e,pos=j;}
}
if(maxn.w<=0) break;
else{
ans-=maxn.w;
used[park][pos]=used[pos][park]=1;
used[maxn.u][maxn.v]=used[maxn.v][maxn.u]=0;
}
}
cout<<"Total miles driven: "<<ans<<endl;
}