20pts WA求助
查看原帖
20pts WA求助
663638
Butterfly_qwq楼主2022/10/22 17:42

rt,照着第一篇TJ打的

#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int ans=inf,tmp,tot,cnt,n,m,p,vis[10001],k[10001],p1[10001],m1[10001][10001],tar[10001][10001];
bool cmp(int a,int b)
{
	return m1[p][a]<m1[p][b];
}
void dfs(int num,int nd)
{
	for(int i=num;i<=cnt;i++)
	{
		if(tot+tmp*k[vis[i]]>=ans)return;
		for(int j=nd;j<=p1[vis[i]];j++)
		{
			if(!k[tar[vis[i]][j]])
			{
				cnt++;
				vis[cnt]=tar[vis[i]][j];
				tmp-=m1[vis[cnt]][tar[vis[cnt]][1]];
				tot+=m1[vis[i]][vis[cnt]]*k[vis[i]];
				k[vis[cnt]]=k[vis[i]]+1;
				dfs(i,j+1);
				tot-=m1[vis[i]][vis[cnt]]*k[vis[i]];
				k[vis[cnt]]=0;
				tmp+=m1[vis[cnt]][tar[vis[cnt]][1]];
				cnt--;
			}
			nd=1;
		}
	}
	if(cnt==n)
	{
		if(tot<ans)ans=tot;
		return;
	}
}
int main()
{
	int u,v,w;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)m1[i][j]=inf;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		if(m1[u][v]<w)continue;
		if(m1[u][v]==inf)
		{
			tar[u][++p1[u]]=v;
			tar[v][++p1[v]]=u;
			m1[u][v]=w;
			m1[v][u]=w;
		}
	}
	for(int i=1;i<=n;i++)
	{
		p=i;
		sort(tar[i]+1,tar[i]+1+p1[i],cmp);
		tmp+=m1[i][tar[i][1]];
	}
	for(int i=1;i<=n;i++)
	{
		tot=0;
		cnt=1;
		vis[1]=i;
		tmp-=m1[i][tar[i][1]];
		k[i]=1;
		dfs(1,1);
		k[i]=0;
		tmp+=m1[i][tar[i][1]];
	}
	cout<<ans;
	return 0;
}
2022/10/22 17:42
加载中...