[USACO2.1] 健康的荷斯坦奶牛 dfs

最新推荐文章于 2025-02-27 20:48:09 发布

原创最新推荐文章于 2025-02-27 20:48:09 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#dfs

USACO 同时被 2 个专栏收录

57 篇文章

订阅专栏

基本算法-dfs

13 篇文章

订阅专栏

本文介绍了一个USACO2.1级别的编程问题——健康荷斯坦奶牛，并提供了一种使用深度优先搜索算法的解决方案。通过对输入数据进行处理并运用递归函数来寻找最优解，该方案能够有效地解决这一问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题：https://www.luogu.org/problemnew/show/P1460

//USACO2.1 健康的荷斯坦奶牛
//by dadatu
#include<iostream>
#include<queue>
#include<cstdio>
#include<cstring>
using namespace std;

int kind[1010][1010],a[1010],cur[1010],ans[1010],v[1010],visit[1010];
int n,m,minn=0x3f3f3f3f,cnt=0;

bool judge()
{
	bool f=true;
	if (cnt>=minn) return !f;
	for (int i=0;i<n;i++)
		if (cur[i]<v[i]) f=false;
	return f;

}
void dfs()
{
	if (cnt>=minn) return;//剪枝，如果已经超过当前最优解就不需要再继续搜了
	if (judge())//替换
	{
		minn=cnt;
		for (int i=0;i<cnt;i++) ans[i]=a[i];
	}
	for (int i=0;i<m;i++)
	{
		if (!visit[i]&&i>=a[cnt-1])//剪枝，从小到大排
		{
			visit[i]++;a[cnt++]=i;
			for (int j=0;j<n;j++) cur[j]+=kind[i][j];
			dfs();
			for (int j=0;j<n;j++) cur[j]-=kind[i][j];
			cnt--;visit[i]=0;//回溯
		}
	}
}

int main()
{
	scanf("%d",&n);
	for (int i=0;i<n;i++) scanf("%d",&v[i]);
	scanf("%d",&m);
	for (int i=0;i<m;i++)
		for (int j=0;j<n;j++) scanf("%d",&kind[i][j]);
	dfs(); 
	printf("%d",minn);
	for (int i=0;i<minn;i++) printf(" %d",ans[i]+1);
	printf("\n");
	return 0;
}