阿里笔试 4.29 第一题分水果

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

int max_Find(vector<int> apple, vector<int> man){
    int count = 0;
    //开始发水果     while(1){
        for(int i = 0; i < man.size(); i++){ //每个人按照选择的种类发放一个水果             if(apple[man[i]] == 0){ //某种水果分配完毕，结束分配
                return count;
            }
            else{
                apple[man[i]]--;
            }        
        }
        count++; //每分配一次每个人得到的水果个数加一
    }
}
void dfs_Choice(vector<int> &apple, vector<int> &man, int k, int &max){
    if(k >= man.size()){   //每个人都选择完毕
        int tmp = max_Find(apple, man) + 1;  //分配剩下的水果并求得当前选择下的每人的最大水果数，此前每人已经分配了一个水果         max = tmp > max ? tmp : max;
        return;
    } 
    //第 k 个人选择水果种类
    for(int i = 0; i < apple.size(); i++){
        if(apple[i] == 0){  //该种水果已经没有了，选择下一种
            continue;
        }
        man[k] = i;//第n个人选第 i 种水果         apple[i]--;
        dfs_Choice(apple, man, k+1, max); //第k+1个人选择水果种类并得到一个该类水果         apple[i]++;  //回溯
        man[k] = -1;
    }
    
}

int main(){
    int T = 1;
    cin >> T;        //循环次数
    int n,m;         //n种水果分给m个人
    int count = 0;
    vector<int> apple;  //存放每一种水果的个数 
    vector<int> man;  //存放每个人选择的水果种类
    while(count < T){
        cin >> n >> m;
        int total = 0; 
        apple.assign(n,0);
        man.assign(m,-1);
        for(int i = 0; i < n; i++){
            cin >> apple[i];
            total += apple[i];
        }
        if(total < m) {        //水果不够分,输出0
            cout << 0 << endl;
            continue;
        } 

        int max = 0;  //保存每个人能分配的最大水果数
        dfs_Choice(apple, man, 0, max); //从第 0 个人开始选择水果种类
        
        cout << max << endl;
        count++;
    }
    return 0;
}