CodeM 第二题 可乐

小美和小团最近沉迷可乐。可供TA们选择的可乐共有k种，比如可口可乐、零度可乐等等，每种可乐会带给小美和小团不同的快乐程度。
TA们一共要买n瓶可乐，每种可乐可以买无限多瓶，小美会随机挑选其中的m瓶喝，剩下的n-m瓶小团喝。
请问应该如何购买可乐，使得小美和小团得到的快乐程度的和的期望值最大？
现在请求出购买可乐的方案。

输入描述:

第一行三个整数n，m，k分别表示要买的可乐数、小美喝的可乐数以及可供选择的可乐种数。
接下来k行，每行两个整数a，b分别表示某种可乐分别给予小美和小团的快乐程度。
对于所有数据，1 <= n <= 10,000, 0 <= m <= n, 1 <= k <= 10,000, -10,000 <= a, b <= 10,000

输出描述:

一行k个整数，第i个整数表示购买第i种可乐的数目。
如果有多解，请输出字典序最小的那个。
对于两个序列 a1, a2, ..., ak, b1, b2, ..., bk，a的字典序小于b，当且仅当存在一个位置i <= k满足：
ai < bi且对于所有的位置 j < i，aj = bj；

#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;

int main(){
    int n = 1;
    int m = 0;
    int k = 1;
    cin >> n >> m >> k;
    int ind = 0;
    vector<pair<int,int>> source;
    for(int i = 0; i < k; i++){
        int a = 1;
        int b = 1;
        scanf("%d",&a);
        scanf("%d",&b);
        pair<int,int> par;
        par.first = a;
        par.second = b;
        source.push_back(par);
        double t = source[i].first * m + source[i].second * (n - m);
        double rain = source[ind].first * m + source[ind].second * (n - m);
        if (t >= rain){
            ind = i;
        }
    }

    for (int j = 0; j < k; ++j) {
        if (j == ind){
            cout<<n;
        }else{
            cout<<0;
        }
        if(j != k -1){
            cout << (" ");
        }
    }

    return 0;
}