Educational Codeforces Round 117 (Rated for Div. 2) E. Messages

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;
}