DFS爆了两发tle
查看原帖
DFS爆了两发tle
1229738
CyansQwQ楼主2024/5/13 23:56

刚学DFS,求助大佬

// Problem: P1460 [USACO2.1] 健康的荷斯坦奶牛 Healthy Holsteins
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P1460
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
using namespace std;
const int N=30;
int v,g,V[N],sum[N],res=0x3f3f3f3f;
vector<int>G[N];
int path[N];//记录状态
bool vis[N];//判重
vector<int>ip;
bool check()
{
	for(int i=1;i<=v;i++)
	   if(sum[i]<V[i])return false;
	return true;
}
void dfs(int u,int cnt)
{
	if(check())
	{
		if(res>cnt)//更改答案
		{
			res=cnt;
			ip.clear();
			for(int i=1;i<=u-1;i++)ip.push_back(path[i]);
		}
		return ;
	}
    for(int i=1;i<=g;i++)
	    if(!vis[i])
		  {
			path[u]=i;
			vis[i]=true;
			for(int j=1;j<=v;j++)sum[j]+=G[i][j-1];//计算和
			dfs(u+1,cnt+1);//向下搜索
			for(int j=1;j<=v;j++)sum[j]-=G[i][j-1];//回溯
			vis[i]=false;
		  }
}
int main()
{
	ios::sync_with_stdio(false);cin.tie(0),cout.tie(0);
	cin>>v;
	for(int i=1;i<=v;i++)cin>>V[i];
	cin>>g;
	for(int i=1;i<=g;i++)
	    for(int j=1,x;j<=v;j++)
	        {cin>>x;G[i].push_back(x);}
	dfs(1,0);
	cout<<res<<" ";
	for(int i=0;i<ip.size();i++)cout<<ip[i]<<" ";
	return 0;
}
2024/5/13 23:56
加载中...