A. Glory Addicts codeforces 1738A

Problem - A - Codeforces

题目大意：有n个数字，分为两种，用0，1标识，要求将这些数字重新排序使得他们的和最大，求和的规则为：从左到右依次累加，如果当前的数和左边一个数类型不同，就加两倍，否则正常加一倍

思路：我们要让和最大肯定要尽量使相邻的数都是不同类型的数，所以我们就按照一个0型数1个1型数排序，哪个类型的数字更多，就令那个类型的最小的数字排在最前面，因为第一个数不可能×2，如果两个类型的数数量相等，就找出两个类型的所有数中最小的那个放在前面，然后后面不同类型的数隔一个位置放一个

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5 + 5;
ll a[N], b[N], op[N];
bool cmp(ll a, ll b)
{
	return a > b;
}
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		int n;
		scanf("%d", &n);
		for (int i = 1; i <= n; i++)
		{
			scanf("%lld", &op[i]);
		}
		int cnt1 = 0, cnt2 = 0;//既作为两个数组的指针，又统计了两种数字的个数
		for (int i = 1; i <= n; i++)
		{
			ll x;
			scanf("%lld", &x);
			if (op[i])
			{
				a[++cnt1] = x;//先将所有数字分为a，b两个数组
			}
			else
			{
				b[++cnt2] = x;
			}
		}
		sort(a + 1, a + cnt1 + 1, cmp);
		sort(b + 1, b + cnt2 + 1, cmp);//先把两个数组按从大到小的顺序排序，大数优先*2
		ll sum = 0;
		if (cnt1 < cnt2)
		{//第一类数字更少
			sum += b[cnt2];//第二类数字中最小的放在最前面
			for (int i = 1; i <= cnt1; i++)
			{
				sum += a[i] * 2 + b[i] * 2;//后面cnt1个数字都能*2
			}
			for (int i = cnt1 + 1; i <= cnt2 - 1; i++)
			{
				sum += b[i];//剩下的都是1倍
			}
		}
		else if (cnt1 > cnt2)
		{//第二类数字更少
			sum += a[cnt1];
			for (int i = 1; i <= cnt2; i++)
			{
				sum += a[i] * 2 + b[i] * 2;
			}
			for (int i = cnt2 + 1; i <= cnt1 - 1; i++)
			{
				sum += a[i];
			}
		}
		else
		{
			if (a[cnt1] > b[cnt2])
			{//两种数字数量相同就找两类数字的最小数哪个最大
				sum += b[cnt2];
				cnt2--;
			}
			else
			{
				sum += a[cnt1];
				cnt1--;
			}
			for (int i = 1; i <= cnt1; i++)
			{//除了第一个数外，其余数字都能*2
				sum += a[i] * 2;
			}
			for (int i = 1; i <= cnt2; i++)
			{
				sum += b[i] * 2;
			}
		}
		printf("%lld\n", sum);
	}
	return 0;
}