RC-u4 攻略分队（暴力DFS+判断）-CSDN博客

整体来说的话，这道题思路不是很难，因为数据量比较小（只有六支队伍），所有完全可以利用DFS对所有可能的选组方案进行搜索，选择最优的。难点在于最优方案的选择，在这里我们可以采用记录最优方案的方式把当前最优的方案记录下来，对于后面得到新的方案，将新方案和最优方案进行比较，来判断最优方案需不需要更新，即新的方案是不是更优。

复杂的地方在于判断最优的方案需不需要更新，即新的方案和当前最优方案哪个更好。题目中很明确地给出了五个规则，我们需要按照这五条规则来判断究竟要不要更新最优方案。

1.优先选择每组有至少一个指挥和至少一个工兵的方案

2.如果规则 1 无法满足，则优先选择每组至少有一个指挥的方案

这两个可以视为一个规则，我们在记录最优方案时，定义一个字段rank来表示方案满足规则1、2的哪个规则。如果新的方案满足规则1而最优方案不满足，则更新最优方案。如果新的方案满足规则2而最优方案既不满足规则1也不满足规则2，更更新最优方案。如果新的方案和最优方案满足的规则等级相同，则进行下一步判断。

3.如果所有方案都不满足规则 2，或经过前 2 个规则筛选后，分组方案仍不唯一，则选择两边人数尽可能接近（即两边人数差尽可能小）的方案；

定义字段num表示最优方案的两组人数差，如果新方案的人数差小于最优方案人数差，更新

4.如果满足规则 3 的方案还不唯一，选择讨伐“欧文”的人数大于等于讨伐“亚特”的人数的方案；

定义字段dy，如果新方案满足规则4而最优方案不满足，则更新。如果新方案和最优方案对于规则4的满足程度相同（即都不满足或者都满足），进行下一步判断。

5.如果满足规则 4 的方案还不唯一，选择讨伐“欧文”的队伍编号方案中最小的一个。

对新方案和最优方案存储的队伍信息（欧文组）进行比较（类似字典序），哪个最小选哪个。

完整代码（C++）

#include<bits/stdc++.h>
using namespace std;
#define int long long

//记录队伍信息 
struct node{
	int number;
	int tk;
	int gb;
	int zh;
}a[7];

//记录当前最优方案的信息 
struct res{
	int rank;       //方案满足的基本条件(0、1、2) 
	int num=1e9;    //当前方案的两组人数差 
	bool dy;        //讨伐“欧文”的人数是否大于等于讨伐“亚特”的人数	
	vector<int>ans; //记录讨伐“欧文”的队伍 
}ans;

bool f; //是否满足最基本条件（每组至少一个坦克） 

bool pd[7];//记录当前状态已经选择了哪些队伍讨伐“欧文” 

void dfs(int u){
	if(u>=5)return;//不可能存在6 0分组的情况 
	
	res now;//用来存储当前选择的方案的信息，用来和当前最优方案进行比较 
	
	//记录分组的信息，tk、gb、zh、num分别表示两个组的坦克、工兵、指挥、总人数 
	int tk1=0,tk2=0,gb1=0,gb2=0,zh1=0,zh2=0,num1=0,num2=0;
	
	for(int i=1;i<=6;i++){ 
		if(pd[i]){  //如果选择了讨伐"欧文" 
			now.ans.push_back(i);
			num1+=a[i].number;
			tk1+=a[i].tk;
			gb1+=a[i].gb;
			zh1+=a[i].zh; 
		}else{
			num2+=a[i].number;
			tk2+=a[i].tk;
			gb2+=a[i].gb;
			zh2+=a[i].zh; 
		}
	}
	 
	if(tk1&&tk2){  
		f=1; //满足了基本条件（两组至少一个坦克）
		if(zh1&&gb1&&zh2&&gb2){ 
			now.rank=1; //满足条件1 
		}else if(zh1&&zh2){
			now.rank=2; //满足条件2 
		}
		
		int diff=abs(num1-num2); 
		now.num=diff; //计算两组人数差 
		
		//判断讨伐“欧文”组人数是否大于等于讨伐“亚特”人数 
		if(num1>=num2) 
			now.dy=1;
		else 
			now.dy=0;
		
		//和当前最优方案比较，判断是否更新当先最优方案	
		if(now.rank==1&&ans.rank!=1){
			//如果当前方案满足规则1，最优方案不满足，更新最优方案 
			ans=now;
		}else if(now.rank==2&&!ans.rank){
			//如果当前方案满足规则2，最优方案规则1、2都不满足，更新最优方案 
			ans=now;
		}else{
			//如果当前方案和最优方案满足的规则1、2相同，进一步判断规则3 
			if(now.num<ans.num)ans=now; //人数差更少，更新 
			else if(now.num==ans.num){  
				if(now.dy==1&&ans.dy==0)ans=now; //人数差相同，如果当前方案满足规则4，最优不满足，更新最优 
				else if(now.dy==ans.dy){ //规则四也分辩不了，选择讨伐“欧文”的队伍编号方案中最小的
					for(int i=0;i<min(now.ans.size(),ans.ans.size());i++){
						//类似字典序 
						if(now.ans[i]<ans.ans[i]){
							ans=now;
							break;
						}else if(now.ans[i]>ans.ans[i]){
							break;
						}
					}
				}
			}
		}
	}
	
	//正常的暴力遍历过程 
	for(int i=1;i<=6;i++){
		if(pd[i])continue;
		pd[i]=1;
		dfs(u+1);
		pd[i]=0;	
	}
}

void solve(){
	for(int i=1;i<=6;i++)cin>>a[i].number;
	//记录队伍信息 
	for(int i=1;i<=6;i++){
		string s;cin>>s;
		a[i].tk=s[0]-'0';
		a[i].gb=s[1]-'0';
		a[i].zh=s[2]-'0';
	}
	
	dfs(0);
	
	//如果不满足最基础条件（每组至少一个坦克） 
	if(!f){
		cout<<"GG"<<endl;
		return;
	}
	
	
	bool flag=0;//末尾不能有空格，用来控制输出 
	
	
	//输出欧文队（答案记录的队伍） 
	for(int i=0;i<ans.ans.size();i++){
		//只有队伍人数不为0才存在 
		if(a[ans.ans[i]].number){
			if(flag==0){
				flag=1;
				cout<<ans.ans[i];
			}else cout<<" "<<ans.ans[i];
		}
	}
	cout<<endl;
	
	
	//输出亚特队（答案中没出现的且存在的队伍） 
	flag=0;
	for(int i=1;i<=6;i++){
		bool flag1=false;//用来判断答案中有没有出现 
		for(int j=0;j<ans.ans.size();j++){
			if(i==ans.ans[j]){
				flag1=true;
				break;
			}
		}
		
		if(!flag1&&a[i].number){
			if(flag==0){
				flag=1;
				cout<<i;
			}else{
				cout<<" "<<i;
			}
		}
	}
}


signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	
	int t=1;
	//cin>>t;
	while(t--){
		solve();
	}
	
	return 0;
}