CodeForces-1490E-Accidental Victory

题目大意

每个人有一个数值a[i]，每一个玩家可以任意选取一个玩家j对战，如果a[i]>a[j],那么i胜利并得到j的数值a[j],每两人之间只交战一次，如果a[i]=a[j]，那么i和j都有一半的机会胜利。问有多少人能成为最后的冠军？（打赢剩下的n-1个对手）

思路

假设a[i]从小到大排列：

dp[i]表示前i个玩家累积的数值a[i]总和，那么dp[i]-a[i]就表示前i-1个玩家累积的数值总和，亦可以理解为第i-1个玩家打赢了他前面的所有玩家最终获得的数值a[i]总和（第i-1个人一定可以打赢前面的i-2个人,因为a[i]从小到大排列），如果dp[i]-a[i]<a[i]说明第i-1个人即使打遍前面的所有人最后也打不赢第i个人，这是第i-1个玩家成为冠军的充分必要条件。这里倒序枚举dp[i]-a[i]，直到找到第一个dp[i]-a[i]<a[i]的位置中断(设maxx=a[i])，说明一个玩家要成为冠军的条件就是a[i]>=maxx,这样就可以找到所有满足条件的值。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+50;
ll n,m,T,a[maxn],dp[maxn],b[maxn];
int main(){
	cin>>T;
	while(T--){
		cin>>n;
		for(int i=1;i<=n;i++)cin>>a[i],b[i]=a[i];
		sort(a+1,a+1+n);
		for(int i=1;i<=n;i++){
			dp[i]=dp[i-1]+a[i];
		}
		ll maxx=0;
		for(int i=n;i>=1;i--){
			if(dp[i]-a[i]<a[i]){
				maxx=a[i];
				break;
			}
		}
		vector<int>s;
		for(int i=1;i<=n;i++){
			if(maxx<=b[i]){
				s.push_back(i);
			}
		}
		cout<<s.size()<<endl;
		for(int i=0;i<s.size();i++)cout<<s[i]<<' ';cout<<endl;
	}
	return 0;
}