1362. 健康的荷斯坦奶牛难度: 一般 / 二进制枚举

Posted 2021-12-12 辉小歌

https://www.acwing.com/problem/content/1364/
二进制枚举即可，取一个最小值。

#include<bits/stdc++.h>
using namespace std;
const int N=30;
int a[N],w[N][N],n,m;
vector<int>ans;
int main(void)
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>a[i];
    cin>>m;
    for(int i=0;i<m;i++)
        for(int j=0;j<n;j++) cin>>w[i][j];
    for(int i=0;i<(1<<m);i++)
    {
        int b[N]={0};
        vector<int>temp;
        for(int j=0;j<m;j++)
        {
            if(i>>j&1)
            {
                for(int z=0;z<n;z++) b[z]+=w[j][z];
                temp.push_back(j);
            }
        }
        bool flag=true;
        for(int j=0;j<n;j++) if(b[j]<a[j]) flag=false;
        if(flag)
        {
            sort(temp.begin(),temp.end());
            if(temp.size()<ans.size()||!ans.size()) ans=temp;
            if(temp.size()==ans.size()&&ans>temp) ans=temp;
        }
    }
    cout<<ans.size();
    for(int i=0;i<ans.size();i++) cout<<" "<<ans[i]+1;
    return 0;
}