题目描述:
农民 John 以拥有世界上最健康的奶牛为傲。他知道每种饲料中所包含的牛所需的最低的维他命量是多少。请你帮助农夫喂养他的牛,以保持它们的健康,使喂给牛的饲料的种数最少。
给出牛所需的最低的维他命量,输出喂给牛需要哪些种类的饲料,且所需的饲料剂量最少。
维他命量以整数表示,每种饲料最多只能对牛使用一次,数据保证存在解。
输入格式:
第一行一个整数 v,表示需要的维他命的种类数。
第二行 v 个整数,表示牛每天需要的每种维他命的最小量。
第三行一个整数 g,表示可用来喂牛的饲料的种数。
下面 g 行,第 n 行表示编号为 n饲料包含的各种维他命的量的多少。
输出格式
输出文件只有一行,包括牛必需的最小的饲料种数 $p$;后面有 $p$ 个数,表示所选择的饲料编号(按从小到大排列)。
如果有多个解,输出饲料序号最小的(即字典序最小)。
样例 1
样例输入 1
4
100 200 300 400
3
50 50 50 50
200 300 200 300
900 150 389 399
样例输出 1
2 1 3
提示
【数据范围】
对于 100% 的数据,1< v < 25,1<g< 15。
输入的所有整数在 [1,1000] 范围内。
思路:DFS全排列
代码献上(DFS拓宽版)
#include <iostream>
using namespace std;
int n;
//维他命的种类数
int m;
//饲料的种类
int q[1000];
//牛每天需要的每种维他命的最小量。
int qq[1000][1000];
//每种饲料包含的各种维他命的量的多少
int w[1000];
//变动的排列--对于每一次递归储存的数不同--简称全排列
int ww[1000];
//对最优结果的保存
int min1 = 100000000;
//最小个数达到维他命的满足
bool pd(int x)
{
for (int i = 1; i <= m; i++)
{
int sum = 0;
for (int j = 1; j <= x; j++)
sum += qq[w[j]][i];
//一竖列求和
if (sum < q[i]) return false;
//不满足最低维他命直接退出
}
return true;
}
//判断函数
void DFS(int t, int s)
{
if (t > n) //截止条件
{
if (pd(s)) //判断是否满足最小维他命
{
if (s < min1) //判断满足条件后保证最小值
{
min1 = s; //保存最小值
for (int i = 1; i <= min1; i++)
{
ww[i] = w[i];
}
//对饲料种类的储存
}
}
return;
//结束
}
w[s + 1] = t;
//将下标储存入数组中
DFS(t + 1, s + 1);
//下标增加,个数增加开始递归
//w[s + 1] = 0;这一步可要可不要不影响结果
DFS(t + 1, s );
//下标增加不变,个数由加一转为不变
}
int main()
{
cin >> m;
//维他命的种类数的输入
for (int i = 1; i <= m; i++)cin >> q[i];
//维他命的具体量的输入
cin >> n;
//饲料的量的输入
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= m; j++)
{
cin >> qq[i][j];
}
}
DFS(1, 0);
//全排列函数
cout << min1 << ' ';
//最小个数输出
for (int i = 1; i <= min1; i++)cout << ww[i] << ' ';
//具体储存的对应下标
return 0;
}