其他分享
首页 > 其他分享> > UVA1537 Picnic Planning

UVA1537 Picnic Planning

作者:互联网

度限制最小生成树。

先不加连跟的边,跑出其他的x个最小生成森林。

用最小代价的边链接根和森林,得到x度时的最小生成树。

如果度限制小于x,无解

如果度大于x,我们需要继续加链接根的边。

每一次枚举所有可能可以加入的边,找到加入后形成的环上不与根链接的边中最长的那一条删掉。在所有能加入的边中取最优的。这样我们可以构造出x+1度的最小生成树

一直构造即可。

#include<bits/stdc++.h>
using namespace std;
#define orz cout<<"lytcltcltcltcltcltcl"<<endl
inline int r(){int s=0,k=1;char c=getchar();while(!isdigit(c)){if(c=='-')k=-1;c=getchar();}while(isdigit(c)){s=s*10+c-'0';c=getchar();}return s*k;}
int t,n,lst,head[10005],cnt,m,dis[10005],fa[10005],ans,b[10005],mini[10005],maxi[105],times,f[10001],rest,valid[10005],root,link[10005],g[105];
map<string,int>mp;
struct node
{
	int from,to,dis,next,link;
	void clear()
	{
		from=to=dis=next=link=0;
	}
}a[10001];
node e[10001];
void add_edge(int from,int to,int dis)
{
	a[++cnt].to=to;
	a[cnt].next=head[from];
	head[from]=cnt;
	a[cnt].dis=dis;
	a[cnt].from=from;
}
bool cmp(node x,node y)
{
	return x.dis<y.dis;
}
int father(int x)
{
	if(fa[x]!=x)fa[x]=father(fa[x]);
	return fa[x];
}
void dfs1(int u)
{
	b[u]=times;
	for(int i=head[u];i;i=a[i].next)
	{
		int v=a[i].to;
		if(!b[v])dfs1(v);
	}
}
void dfs2(int u,int fa,int deep,int k)
{
//	cout<<"dfs"<<u<<endl;
	b[u]=times;
	maxi[u]=deep;
	g[u]=k;
	for(int i=head[u];i;i=a[i].next)
	{
		int v=a[i].to;
		if(v==fa)continue;
		if(link[i])continue;
		if(u!=root)
		{
			if(deep<a[i].dis)dfs2(v,u,a[i].dis,i);
			else dfs2(v,u,deep,k);
		}
		else dfs2(v,u,0,0);
	}
}
void work()
{
	m=r();
	string x,y;
	int z;
	
	for(int i=1;i<=2*m;i++)
	{
		a[i].clear();
		e[i].clear();
	}
	
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y;
		cin>>z;
		if(mp[x]==0)
		{
			n++;
			mp[x]=n;
		}
		if(mp[y]==0)
		{
			n++;
			mp[y]=n;
		}
		int xx=mp[x],yy=mp[y];
		e[i].from=xx;e[i].to=yy;e[i].dis=z;
		add_edge(xx,yy,z);
		add_edge(yy,xx,z);
	}
	
	
	root=mp["Park"];
	for(int i=head[root];i;i=a[i].next)e[(i+1)/2].link=1;
	memset(head,0,sizeof(head));cnt=1;
	sort(e+1,e+m+1,cmp);
	for(int i=1;i<=n;i++)fa[i]=i;
	for(int i=1;i<=m;i++)
	{
		if(e[i].link)continue;
		int x=e[i].from,y=e[i].to;
		int fax=father(x),fay=father(y);
		if(fax==fay)continue;
		fa[fax]=fay;
		ans+=e[i].dis;
		add_edge(e[i].from,e[i].to,e[i].dis);
		add_edge(e[i].to,e[i].from,e[i].dis);
//		cout<<"add"<<e[i].from<<' '<<e[i].to<<' '<<e[i].dis<<endl;
	}
//	cout<<mp["Park"];
	
	
	memset(mini,0x3f,sizeof(mini));
	for(int i=1;i<=n;i++)
	{
		if(!b[i]&&i!=root)
		{
			times++;
			dfs1(i);
		}
	}
	for(int i=1;i<=m;i++)
	{
		if(!e[i].link)continue;
		int x=e[i].from,y=e[i].to;
		if(y==root)swap(x,y);
		if(mini[b[y]]>e[i].dis)
		{
			mini[b[y]]=e[i].dis;
			f[b[y]]=i;
		}
	}
	for(int i=1;i<=times;i++)
	{
		int x=e[f[i]].from,y=e[f[i]].to,z=e[f[i]].dis;
		valid[f[i]]=1;
		add_edge(x,y,z);
		add_edge(y,x,z);
		ans+=mini[i];
//		cout<<"add"<<x<<" "<<y<<" "<<z<<endl;
	}

//	cout<<ans<<endl;	
	lst=ans;
	rest=r();
	for(int ttt=times+1;ttt<=rest;ttt++)
	{
		memset(maxi,0,sizeof(maxi));
		dfs2(root,0,0,0);//find the maxi
//		for(int i=1;i<=n;i++)
//		cout<<"maxi"<<i<<":"<<maxi[i]<<" "<<g[i]<<endl;
		
		int choose_maxi=-1e9,choose_i=0,choose_j=0;
		for(int i=1;i<=m;i++)
		{
			if(!e[i].link)continue;
			if(valid[i])continue;
			int x=e[i].from,y=e[i].to;
			if(y==root)swap(x,y);
			if(choose_maxi<maxi[y]-e[i].dis)
			{
				choose_maxi=maxi[y]-e[i].dis;
				choose_i=g[y];
				choose_j=i;
			}
		}
		
//		cout<<"get"<<choose_maxi<<" "<<e[choose_j].dis<<" "<<a[choose_i].dis<<endl;

		lst-=choose_maxi;
		ans=min(ans,lst);
		link[choose_i]=1;
		link[choose_i^1]=1;
		add_edge(e[choose_j].from,e[choose_j].to,e[choose_j].dis);
		add_edge(e[choose_j].to,e[choose_j].from,e[choose_j].dis);
	}
	
	cout<<"Total miles driven: "<<ans<<"\n";
}
void init()
{
	n=lst=cnt=m=ans=times=root=rest=0;
	memset(b,0,sizeof(b));
	memset(mini,0,sizeof(mini));
	memset(maxi,0,sizeof(maxi));
	memset(head,0,sizeof(head));
	memset(dis,0,sizeof(dis));
	memset(valid,0,sizeof(valid));
	memset(g,0,sizeof(g));
	memset(f,0,sizeof(f));
	memset(fa,0,sizeof(fa));
	memset(link,0,sizeof(link));
	mp.clear();
}
int main()
{
//	freopen("a.in","r",stdin);
//	freopen("b.txt","w",stdout);
	t=r();
	while(t--)
	{
		init();
		work();
		if(t!=0)puts("");
	}
}

标签:cnt,Picnic,int,head,UVA1537,xx,Planning,mp,dis
来源: https://www.cnblogs.com/lytql/p/15224060.html