CF1728A Colored Balls: Revisited

思路

设袋子中只剩下颜色为 \(x\) 的小球，那么其他颜色的球已经被取走过了。

那么取球方式可以分为以下两种：

• 取出两个颜色均不为 \(x\) 且颜色不相同的两个小球。

• 取出一个颜色为 \(x\) 的小球和一个颜色不为 \(x\) 的小球。

先不考虑第二种情况：

即求一个子问题：从 \(n-1\) 种小球中每次取出两个颜色不同的球，使剩下的小球尽量少（从而使最后颜色为 \(x\) 的小球尽量多，同时也符合题意）。

考虑第二种情况：

如果此时剩下的小球大于颜色为 \(x\) 的小球的个数，那么袋子中只剩下颜色为 \(x\) 的小球的方案不存在，反之一定有解。

那么如何求从 \(n-1\) 种小球中每次取出两个颜色不同的球，使剩下的小球尽量少这个问题呢？

每次将两个颜色不同的小球取出即可。

for(int i=1; i<=n; i++)
{
	sum=0;//剩下的小球的个数
	for(int j=1; j<=n; j++)//枚举除颜色为 x 以外的其他小球
	{
		if(j==i)continue;
		sum=abs(sum-cnt[j].x);
		// 前 j-1 种小球剩余的球的个数 sum 与第 j 种小球 cnt[j].x 
		// 取出前 j 种小球剩余的球 sum = abs(sum-cnt[j].x)
	}
	if(sum<cnt[i].x)return cnt[i].id;
	//存在方案使袋子中只剩下颜色为 i 的小球,直接返回
}

\(Code\)

#include<bits/stdc++.h>
using namespace std;
int T;
struct node{
	int x,id;
};
class sovle{
	private:
		int n,ans,sum;
		node cnt[30];
	public:
		int work()
		{
			scanf("%d",&n);
			for(int i=1; i<=n; i++)
			{
				cin>>cnt[i].x;
				cnt[i].id=i;
			}
			for(int i=1; i<=n; i++)
			{
				sum=0;
				for(int j=1; j<=n; j++)
				{
					if(j==i)continue;
					sum=abs(sum-cnt[j].x);
				}
				if(sum<cnt[i].x)return cnt[i].id;
			}
		}
};
int main()
{
	sovle x;
	scanf("%d",&T);
	while(T--)
	{
		cout<<x.work()<<endl;
	}
}

标签：cnt,Balls,颜色,int,小球,取出,剩下,Revisited,CF1728A
来源： https://www.cnblogs.com/dadidididi/p/16674044.html