Educational Codeforces Round 117 (Rated for Div. 2) E. Messages(思维+暴力)
这个题目意思十分的绕,理解题意就十分好写了。具体看代码块
E. Messages
//https://codeforces.com/contest/1612/problem/E
//有n个人,第i个人想要看到消息mi,如果看到就做出贡献1,没有就贡献0
//同时第i个人最多可以看ki条信息
//问:最多投放多少条信息,使得(贡献/信息总数)最大,并输出具体是哪些信息
//原因ki的范围为1<=ki<=20,所以直接暴力枚举ki就行了(具体看注释)
//找出最大的res值,最后输出。
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+5;
int n,mi[N],ki[N];
double mx;
vector<int>ans;
struct node{
int pro,id;//第id种信息的贡献数
}a[N];
bool cmp(node l,node r){return l.pro>r.pro;}
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%d%d",&mi[i],&ki[i]);//读入
for(int k=1;k<=20;k++){//枚举选择投放k条信息
for(int i=1;i<=200000;i++) a[i]={0,i};//初始化第i种信息可能被选取的次数为0
//对于第i个人,总贡献数为本次投放信息数k和自己最多看ki[i]条信息的最小值
//选取概率1/k,则选到mi的概率为min(k,ki[i])/k
for(int i=1;i<=n;i++) a[mi[i]].pro+=min(k,ki[i]);
sort(a+1,a+200001,cmp);//将贡献从大到小排序
vector<int>temp;
double res=0.0;//选中次数
for(int i=1;i<=k;i++){//第k次就找前k个数就行了
res+=a[i].pro;
temp.push_back(a[i].id);
}
//最后肯定要除以k,因为题目是要求最大化阅读各自消息的预期学生人数
//(maximize the expected number of students that read their respective messages)
//简化一下:就是求(选中次数/信息总数)的最大值
res/=k;
if(res>mx){//更新答案
mx=res;
ans=temp;
}
}
printf("%d\n",ans.size());//输出
for(int i=0;i<ans.size();i++) printf("%d%c",ans[i],(i==ans.size()-1)?'\n':' ');
return 0;
}