Submission #1241328

#TimeUsernameProblemLanguageResultExecution timeMemory
1241328GeforgsLet's Win the Election (JOI22_ho_t3)C++20
100 / 100
251 ms4416 KiB
#include <iostream>
#include <iomanip>
#include <vector>
#include <cmath>
#include <algorithm>
#include <set>
#include <queue>
#include <map>
#include <unordered_map>
#include <stack>
#include <bitset>
#include <string>
#include <cstring>
#include <iterator>
#include <random>
#define ll long long
#define ld long double
#define inf (ll)(2*1e18)
#define sort(a) sort(a.begin(), a.end())
#define reverse(a) reverse(a.begin(), a.end())
#define pb push_back
#define endl "\n"
using namespace std;

void solve(){
    ll n, k, i, x, w, cnt, left, j, g;
    ld res=inf, sum;
    cin>>n>>k;
    vector<pair<ll, ll>> a(n);
    vector<vector<ld>> dp(k+1, vector<ld>(k+1, inf));
    multiset<ll> s1;
    for(i=0;i<n;++i){
        cin>>x>>w;
        if(w == -1) w = inf;
        a[i] = {w, x};
        s1.insert(x);
    }
    for(i=0;i<k;++i){
        sum += *s1.begin();
        s1.erase(s1.begin());
    }
    sort(a);
    dp[0][0] = sum;
    for(i=1;i<=k;++i){
        dp[i][0] = 0;
    }
    for(i=0;i<n;++i){
        if(a[i].first == inf or i == k) break;
        multiset<ll> s;
        for(j=i+1;j<n;++j){
            s.insert(a[j].second);
        }
        sum = 0;
        for(j=i+1;j<k;++j){
            sum += *s.begin();
            s.erase(s.begin());
        }
        for(j=k;j>0;--j){
            dp[j][j] = min(dp[j][j], dp[j][j-1] + sum/(ld)(j+1) + (ld)a[i].first/(ld)j);
            for(g=j-1;g>0;--g){
                dp[j][g] = min(dp[j][g] + (ld)a[i].second/(ld)(j+1), dp[j][g-1] + (ld)a[i].first/(ld)g);
            }
            dp[j][0] += (ld)a[i].second/(ld)(j+1);
        }
    }
    for(i=0;i<=k;++i){
        res = min(res, dp[i][i]);
    }
    cout<<fixed<<setprecision(20)<<res<<endl;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    srand(time(nullptr));
    ll t=1;
//    cin>>t;
    for(;t>0;--t){
        solve();
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...