USACO-Section2.1 Healthy Holsteins （深度优先搜索）

2017-8-15

题目描述

1.喂给牛的饲料的种数最少
2.所需的饲料剂量最少
3.输出饲料序号最小的

解答

我用的是dfs求出所有情况
1.最大种数从1开始到n，若当前种数有满足条件的，就是他了，break即可
2.每次k从1开始为了节省数组r的空间
3.set函数！！！

代码

/*
ID: 18795871
PROG: holstein
LANG: C++
*/
#include<iostream>
#include<fstream>
#include<cstring>
using namespace std;
const int M = 25,N = 15;

ifstream fin("holstein.in");
ofstream fout("holstein.out");

int x[M+1],y[N+1][M+1]; //init
int s[M+1]; //sum:now
int r[10000][N+1]; //type:now
bool f[N+1],res=false;
int now,m,n,sum=0,k=1,qq;

bool flag(){ //判断当前是否符合条件
    for (int i=1;i<=m;i++){
        if (s[i]<x[i]) return false;
    }
    return true;
}

int fun(){ //求出当前所需的饲料剂量
    int res=0;
    for (int i=1;i<=m;i++){
       res+=s[i];
    }
    return res;
}

void cal(int j){ 
    for (int i=1;i<=m;i++){
        s[i]+=y[j][i];
    }
}

void mul(int j){ 
    for (int i=1;i<=m;i++){
        s[i]-=y[j][i];
    }
}

void set(int p){
    for (int i=1;i<=k;i++){
        for (int j=0;j<p;j++){
            if (!r[i][j]){
                r[i][j]=r[i-1][j];
            }
        }
    }
}

void dfs(int step,int cnt){
   if (step==n+1){
      k++;
      return ;
   }
   if (cnt==now){
       set(cnt);
       if (flag()){
          int fun_res=fun();
          if (sum==0||(sum!=0&&fun_res<sum)){
             qq=k;
             sum=fun_res;
          }
          res=true;
       }
       k++;
       return ;
   }
   for (int i=step+1;i<=n;i++){
          if (!f[i]){
             cal(i);
             r[k][cnt]=i;
             f[i]=true;
             dfs(i,cnt+1);
             f[i]=false;
             mul(i);
          }
   }
}

void show(int p){
    int i;
    fout<<now<<" ";
    for (i=0;i<now-1;i++){
        fout<<r[p][i]<<" ";
    }
    fout<<r[p][i]<<endl;
}

int main()
{
    int i,j;
    fin>>m;
    for (i=1;i<=m;i++){
        fin>>x[i];
    }
    fin>>n;
    for (i=1;i<=n;i++){
        for (j=1;j<=m;j++){
            fin>>y[i][j];
        }
    }
    for (i=1;i<=n;i++){
        if (res) break;
        now=i;
        k=1;
        memset(s,0,sizeof(s));
        memset(f,false,sizeof(f)); 
        memset(r,0,sizeof(r));
        dfs(0,0);
    }
    show(qq);
    return 0;
}

上面的思路不是很明白了现在？！说一下现在的思路吧！
每种饲料我们可以选择选或者不选，那么在这种情况下我们可以选择深度优先搜索求解：
（1）为了使所求得的序列是最小的，那么我们应该首先选择 ” 选 ” 。
（2）我用p数组标记当前饲料选否！r数组存储当前的选择时各种vit的和，q数组存放最后的答案，当我们来到最后一个饲料的时候，我们就要判断当前的选择是否能够满足我们对vit的要求，即r数组和v数组进行比较。需要注意的是，我们优先要求的是所需饲料种类最少，其次是序列号较小的那一个！
（3）最后输出的时候，如果是最后一个的话，记得输出换行符，否则的话，
输出一个空格。

/*
ID: 18795871
PROG: holstein
LANG: C++
*/
#include <iostream>
#include <fstream>
using namespace std;

ifstream fin("holstein.in");
ofstream fout("holstein.out");

const int M = 25, N = 15;
int v[M+1],x[N+1][M+1],r[M+1];
int vv,gg,cnt;
bool p[N+1],q[N+1];

bool isOK(){
    int i;
    for (i=0;i<vv;i++){
        if (r[i]<v[i]) return false;
    }
    return true;
} 

void dfs(int step,int num){
    if (step==gg){
        if (isOK()){
            if (num<cnt){
                cnt=num;
                for (int i=0;i<gg;i++){
                    q[i]=p[i];
                }
            }
        }
        return ;
    }
    p[step]=1;
    for (int i=0;i<vv;i++){
        r[i]+=x[step][i];
    }
    dfs(step+1,num+1);
    p[step]=0;
    for (int i=0;i<vv;i++){
        r[i]-=x[step][i];
    }
    dfs(step+1,num);
}

int main(){
    int i,j; 
    while (fin>>vv){
        for (i=0;i<vv;i++){
            fin>>v[i];
        }
        fin>>gg;
        for (i=0;i<gg;i++){
            for (j=0;j<vv;j++){
                fin>>x[i][j];
            }
        }
        cnt=N;
        dfs(0,0);
        fout<<cnt<<" ";
        int k=0;
        for (i=0;i<gg;i++){
            if (q[i]){
                k++;
                if (k<cnt) fout<<i+1<<" ";
                else fout<<i+1<<endl;
            } 
        }
    }
    return 0;
}