此题能否用dfs+卡时剪枝瞎搞过掉
查看原帖
此题能否用dfs+卡时剪枝瞎搞过掉
250699
mot1ve楼主2022/12/12 18:24

瞎搞了一个,78分,还能改进一下吗

//假的dfs,vis[i][j]表示在i点体重为j,vis可以回溯 
#include<bits/stdc++.h>
using namespace std;
int n,m,k,ans,idx,cnt;
int flag[10010],head[10010];
int vis[110][110];
struct node{
	int nxt,to,w;
}edge[100010];
void add(int u,int v,int w)
{
	edge[++idx].nxt=head[u];
	edge[idx].to=v;
	edge[idx].w=w;
	head[u]=idx;
}
void dfs(int x,int y)
{
	cnt++;
	if(cnt>1e7)
	{
		cout<<ans;
		exit(0);
	}
	if(vis[x][y])
	return ; 
	if(x==1)
	{
		ans=max(ans,y);
	}
	for(int i=head[x];i;i=edge[i].nxt)
	{
		if(y>edge[i].w)
		continue;
		int v=edge[i].to;
		if(flag[v])//有草 
		{
			for(int j=0;j<=1;j++)//j=0,吃,j=1不吃 
			{
				if(j==0)
				{
					flag[v]=0;
					vis[x][y]=1;
					dfs(v,y+1);
					flag[v]=1;
					vis[x][y]=0;
				}
				if(j==1)
				{
					vis[x][y]=1;
					dfs(v,y);
					vis[x][y]=0;
				}
			}
		}
		else//没草 
		{
			vis[x][y]=1;
			dfs(v,y);
			vis[x][y]=0; 
		}
	}
}
int main()
{
	//freopen("wngdh.in","r",stdin);
	//freopen("answer.out","w",stdout);
	cin>>n>>m>>k;
	for(int i=1;i<=k;i++)
	{
		int x;
		scanf("%d",&x);
		flag[x]=1;
	}
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0);
	cout<<ans;
	return 0;
}
2022/12/12 18:24
加载中...