您的位置:首页 > 其它

USACO 2.1 Healthy Holsteins (DFS)

2015-09-30 19:05 429 查看
#include <stdio.h>
#define DEBUG 1
#define TESTCASES 9
#define MAX_TYPES_VITAMINS 25
#define MAX_TYPES_FEEDS 15
int typesOfVitamins;
int minRequare[MAX_TYPES_VITAMINS + 1];
int typesOfFeeds;
int vitaminOfFeed[MAX_TYPES_FEEDS + 1][MAX_TYPES_VITAMINS + 1];
int currentVitamin[MAX_TYPES_VITAMINS + 1];
int stack[MAX_TYPES_FEEDS + 1];
int top;
int feedsEatten[MAX_TYPES_FEEDS + 1];
int minTypesOfFeedsEatten;

int enough(){
int vitamin;
for (vitamin = 1; vitamin <= typesOfVitamins; vitamin++)
if (currentVitamin[vitamin] < minRequare[vitamin])
return 0;
return 1;
}

void eat(int feed){
if (top >= minTypesOfFeedsEatten)
return;
if (enough()){
minTypesOfFeedsEatten = top;
int i;
for (i = 1; i <= top; i++)
feedsEatten[i] = stack[i];
return;
}
if (feed > typesOfFeeds)
return;
int vitamin;
for (vitamin = 1; vitamin <= typesOfVitamins; vitamin++)
currentVitamin[vitamin] += vitaminOfFeed[feed][vitamin];
stack[++top] = feed;
eat(feed + 1);
for (vitamin = 1; vitamin <= typesOfVitamins; vitamin++)
currentVitamin[vitamin] -= vitaminOfFeed[feed][vitamin];
top--;
eat(feed + 1);
}

int main(){
#if DEBUG
int testCase;
for (testCase = 1; testCase <= TESTCASES; testCase++){
char inputFileName[20] = "inputx.txt";
inputFileName[5] = '1' +  (testCase - 1);
freopen(inputFileName, "r", stdin);
printf("\n#%d\n", testCase);
#endif

scanf("%d", &typesOfVitamins);
int vitamin;
for (vitamin = 1; vitamin <= typesOfVitamins; vitamin++)
scanf("%d", &minRequare[vitamin]);

scanf("%d", &typesOfFeeds);
int feed;
for (feed = 1; feed <= typesOfFeeds; feed++)
for (vitamin = 1; vitamin <= typesOfVitamins; vitamin++)
scanf("%d", &vitaminOfFeed[feed][vitamin]);

for (vitamin = 1; vitamin <= typesOfVitamins; vitamin++)
currentVitamin[vitamin] = 0;
top = 0;
minTypesOfFeedsEatten = MAX_TYPES_FEEDS + 1;

eat(1);

printf("%d ", minTypesOfFeedsEatten);
int i;
for (i = 1; i <= minTypesOfFeedsEatten; i++)
printf("%d ", feedsEatten[i]);
printf("\n");

#if DEBUG
}
#endif
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息