最简单的搜索为什么都过不了样例
查看原帖
最简单的搜索为什么都过不了样例
250699
mot1ve楼主2020/9/12 11:04

是细节出问题了还是思路?

#include<bits/stdc++.h>
using namespace std;
int n,m,k,ans=0x3f3f3f3f;
int a[110][110],head[110],own[110];
int color[15];
bool vis[110];
struct node{
	int nxt,to,w;
}edge[100010];
int idx;
void add(int u,int v,int w)
{
	edge[++idx].nxt=head[u];
	edge[idx].to=v;
	edge[idx].w=w;
	head[u]=idx;
}
bool check()
{
	for(int i=0;i<=n-1;i++)
	{
		if(!own[i])
		return 0;
	}
	return 1;
}
void dfs(int x,int dis)//dis记录已经走过的距离 
{
	if(check())
	ans=min(ans,dis);
	for(int i=head[x];i;i=edge[i].nxt)
	{
		int v=edge[i].to;
		if(vis[v])//不能重复经过一个点 
		continue;
		vis[v]=1;
		own[color[v]]=1; 
		dfs(v,dis+edge[i].w);
		own[color[v]]=0;
		vis[v]=0;
	}
}
int main()
{
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&color[i]);
	}
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
	}
	for(int i=1;i<=n;i++)
	{
		memset(own,0,sizeof(own));
		memset(vis,0,sizeof(vis));
		vis[i]=1;
		own[color[i]]=1;
		dfs(i,0);//枚举从每个点开始走
		vis[i]=0;
		own[color[i]]=0; 
	}
	cout<<ans;
	return 0;
}
2020/9/12 11:04
加载中...