Submission #806193

#	Time	Username	Problem	Language	Result	Execution time	Memory
806193		vjudge1	Let's Win the Election (JOI22_ho_t3)	C++14	10 / 100	1 ms	384 KiB

Main

#include<bits/stdc++.h>
#define fi first
#define se second
#define ll long long
#define ld long double
using namespace std ;
const int N = 500 ;
bool flag1, flag2 ;
int n, k ;
pair<pair<ld, ld>, int> p[N + 1] ;
signed main()
{
    ios_base::sync_with_stdio( 0 ) ;
    cin.tie( 0 ) ;
    cout.tie( 0 ) ;
    cin >> n ;
    cin >> k ;
    for(int i = 1 ; i <= n ; i++)
    {
        cin >> p[i].fi.fi >> p[i].fi.se ;
        p[i].se = i ;
        if(p[i].fi.se != -1)
            flag1 = 1 ;
        if(p[i].fi.se != -1 && p[i].fi.se != p[i].fi.fi)
            flag2 = 1 ;
    }
    if(!flag1)
    {
        sort(p + 1, p + n + 1) ;
        ld ans = 0 ;
        for(int i = 1 ; i <= k ; i++)
            ans += p[i].fi.fi ;
        cout << fixed << setprecision(9) << ans ;
        return 0 ;
    }
    if(!flag2)
    {
        ld ans = 1e9 ;
        vector<pair<pair<ld, ld>, int>> v_pos, v_neg ;
        for(int i = 1 ; i <= n ; i++)
        {
            if(p[i].fi.se == -1)
                v_neg.push_back(p[i]) ;
            else
                v_pos.push_back({{p[i].fi.se, p[i].fi.fi}, i}) ;
        }
        sort(v_pos.begin(), v_pos.end()) ;
        sort(v_neg.begin(), v_neg.end()) ;
        for(int i = -1 ; i < min(k, (int)v_pos.size()) ; i++)
        {
            int now = i + 1, uk1 = i + 1, uk2 = 0 ;
            ld sum = 0, cnt = 1 ;
            for(int j = 0 ; j <= i ; j++)
            {
                sum += v_pos[j].fi.fi / cnt ;
                cnt++ ;
            }
            while(now < k)
            {
                ld mn = 1e9 ;
                if(uk1 < v_pos.size())
                    mn = v_pos[uk1].fi.se ;
                if(uk2 < v_neg.size())
                    mn = min(mn, v_neg[uk2].fi.fi) ;
                sum += mn / cnt ;
                if(uk1 < v_pos.size() && mn == v_pos[uk1].fi.se)
                    uk1++ ;
                else
                    uk2++ ;
                now++ ;
            }
            ans = min(ans, sum) ;
        }
        cout << fixed << setprecision(9) << ans ;
        return 0 ;
    }
    return 0 ;
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:61:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::pair<long double, long double>, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   61 |                 if(uk1 < v_pos.size())
      |                    ~~~~^~~~~~~~~~~~~~
Main.cpp:63:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::pair<long double, long double>, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   63 |                 if(uk2 < v_neg.size())
      |                    ~~~~^~~~~~~~~~~~~~
Main.cpp:66:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<std::pair<long double, long double>, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   66 |                 if(uk1 < v_pos.size() && mn == v_pos[uk1].fi.se)
      |                    ~~~~^~~~~~~~~~~~~~

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