题目大意
每个人有一个数值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;
}