好题分享–小A的皇室战争卡组

题目描述

小 A 很喜欢玩皇室战争这款游戏。皇室战争的卡牌种类共有 3 种，分别为部队、建筑和法术。一个卡组会携带 88 张卡牌，一个合理的卡组会包含 0∼1 张建筑牌、 1∼3 张法术牌，部队牌则没有限制。每张卡牌都有自己独立的等级 ai​ 和种类 ci​。

现在小 A 想和 小 B 进行友谊战，因此需要来构建出战卡组。小 A 共有 n 张卡牌，卡牌的类型与等级已知，且 小 B 的每张出战卡牌的等级已知。由于小 A 对自己的水平非常自信，因此只要 出战卡牌的平均等级 ≥ 对手出战卡牌的平均等级 −2，且卡组为一个合理的卡组，则他就能确保取胜。

现在请你帮小 A​ 想想，他可以确保取胜吗？

输入格式

第一行一个整数 T (T≤103) 表示数据组数。

每组第一行输入一个正整数 n (8≤n≤105) ，代表 小A 的卡牌总数，保证 ∑�≤105∑n≤105

每组第二行输入 n 个正整数，代表每张卡牌的种类  ci​ (1≤ci​≤3) ，其中种类 1 代表部队卡，种类 2代表建筑卡，种类 3 代表法术卡。

每组第三行输入 n 个正整数，代表每张卡牌的等级 ai​ (1≤ai​≤15)

每组第四行输入 8 个正整数，代表 小 B 每张卡牌的等级 xi​ (1≤xi​≤15)。

输出格式

共 T 行，每行一个字符串。如果 小A 可以取胜则输出 Yes，否则输出 No

输入输出样例

输入 

3
10
2 3 3 3 1 1 1 1 1 1 
7 8 9 1 2 3 4 5 6 6
13 13 13 13 3 3 3 3
10
2 2 3 1 1 1 1 1 1 1
10 9 2 2 2 2 2 2 2 2
5 5 5 5 5 5 5 6
8
1 1 1 1 1 1 1 1
15 15 15 15 15 15 15 15
1 1 1 1 1 1 1 1

输出 

Yes
No
No

思路：

1.先用结构体数组记录数据，然后按建筑牌，部队牌，法术牌分开排序

2.依题意，建筑牌有两种选择0张或1张，法术牌有三种选择1张、2张或3张，部队牌就用8张减去上面两种选择即可

3.由于我们已经排序过了，所以每次都从下标为0开始取

#include <unordered_map>
#include <cstring> 
using namespace std;
struct node{
	int x,y;
};

int T,n,x;
double average;//记录b的等级 
int bsum;
int a[100010],b[100010],c[100010];
node q[100010];
int p[100010];

bool compare(int a,int b){
	return a>=b;
}

int main()
{
	cin>>T;
	for(int i=0;i<T;i++){
		memset(p,0,sizeof p);
		cin>>n;
		double maxs=-1e9;
		for(int j=0;j<n;j++){
			cin>>q[j].x;
			p[q[j].x]++;
		}
		for(int j=0;j<n;j++) cin>>q[j].y;
        for(int j=0;j<8;j++) cin>>x,bsum+=x;
        if(p[3]==0){
			cout<<"No"<<endl;
			continue;
		}
		for(int j=0;j<n;j++){
			if(q[j].x==1) a[--p[1]]=q[j].y;
			else if(q[j].x==2) b[--p[2]]=q[j].y;
			else if(q[j].x==3) c[--p[3]]=q[j].y;
		}
        average=bsum/8-2;
        sort(a,a+n,compare);
        sort(b,b+n,compare);
        sort(c,c+n,compare);
        for(int j=0;j<=1;j++){
        	for(int k=1;k<=3;k++){
        		int asum=0;
        		for(int o=0;o<8-k-j;o++) asum+=a[o];
        		for(int x=0;x<j;x++) asum+=b[x];
        		for(int y=0;y<k;y++) asum+=c[y];
        		maxs=max(maxs,asum*1.0/8);
			}
		}
		if(maxs>=average) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}