Submission #734934

#	Time	Username	Problem	Language	Result	Execution time	Memory
734934		Josia	Let's Win the Election (JOI22_ho_t3)	C++17	100 / 100	332 ms	2388 KiB

Main

#include <bits/stdc++.h>

using namespace std;

#define int int64_t




signed main() {
    cin.tie(0);
    ios_base::sync_with_stdio(0);

    int n, K; cin >> n >> K;
    vector<array<int, 3>> inputA(n);
    vector<array<int, 3>> inputB(n);


    for (int i=0; i<n; i++) {
        int a, b; cin >> a >> b;

        if (b == -1) b = 1e10;

        inputA[i]={a, b, i};
        inputB[i]={b, a, i};
    }

    sort(inputA.begin(), inputA.end());
    sort(inputB.begin(), inputB.end());

    // vector<bool> topK(n, 0);
    // for (int i = 0; i<K; i++) {
    //     topK[inputA[i][2]] = 1;
    // }




    double res = 1e10;

    array<array<double, 505>, 505> dp;
    for (int i=0; i<K; i++) {
        for (auto &a: dp) for (auto &b: a) b = 1e10;
        dp[0][0] = 0;
        for (int j=0; j<n; j++) {
            for (int k=0; k<=n; k++) {
                // cerr << dp[j][k] << " ";
                if (dp[j][k] == 1e10) continue;

                int colabs = j-k;

                int thisColab = 0;
                if (colabs < i) {
                    thisColab = inputB[j][0];
                }

                dp[j+1][k] = min(dp[j+1][k], dp[j][k]+(double)thisColab/(double)(colabs+1));
                dp[j+1][k+1] = min(dp[j+1][k+1], dp[j][k]+(double)inputB[j][1]/(double)(i+1));
            }
            // cerr << "\n";
        }
        // cerr << "\n";
        res = min(res, dp[n][K-i]);
    }



    // for (int colabs = 0; colabs<=K; colabs++) {
    //     for (int inTopK = 0; inTopK<=K; inTopK++) {
    //         double tmp = dp[n][inTopK][colabs];
    //         cerr << tmp << "\n";

    //         if (tmp == 1e10) continue;

    //         for  (int i = 0; i<K+inTopK; i++) {
    //             tmp += inputA[i][0];
    //         }

    //         res = min(res, tmp);
    //         cerr << "-" << tmp << "\n";
    //     }
    // }


    cout.precision(100);
    cout << res << "\n";

    return 0;
}

• Subtask 1: 5 / 5
• Subtask 2: 5 / 5
• Subtask 3: 11 / 11
• Subtask 4: 12 / 12
• Subtask 5: 33 / 33
• Subtask 6: 11 / 11
• Subtask 7: 23 / 23