제출 #1147923

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1147923		andrewp	Nile (IOI24_nile)	C++20	19 / 100	21 ms	5052 KiB

nile

//Dedicated to my love, ivaziva
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define pii pair<int,int>
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define sz(a) ((int)a.size())
const int N=1e5+5;

int n, q;
int w[N], a[N], b[N], e[N];

vector<ll> calculate_costs(vector<int> W, vector<int> A, vector<int> B, vector<int> E) {
    n=sz(W), q=sz(E);
    for (int i=1; i<=n; i++) 
        w[i]=W[i-1], a[i]=A[i-1], b[i]=B[i-1];
    for (int i=1; i<=q; i++) 
        e[i]=E[i-1];
    if (q<=5&&n<=2000&&W==vector<int>(n, 1)) { 
        if (n%2==0) { 
            ll ans=0;   
            for (int i=1; i<=n; i++) 
                ans+=b[i];
            return (vector<ll>(q, ans)); 
        }   
        else {
            ll sumb=0, ans=(ll)(1e18);
            for (int i=1; i<=n; i++) 
                sumb+=b[i];
            for (int i=1; i<=n; i++) 
                ans=min(ans, a[i]+sumb-b[i]); 
            return (vector<ll>(q, ans)); 
        }
    } 
    vector<int> sub2;
    for (int i=0; i<n; i++) sub2.pb(i+1);
    if (q<=5&&W==sub2) {
        if (n%2==0) {  
            ll ans=0;   
            for (int i=1; i<=n; i++) 
                ans+=b[i];
            return (vector<ll>(q, ans));
        }   
        else { 
            vector<ll> ret;         
            for (int i=1; i<=q; i++) {
                if (e[i]>=2) {
                    ll sumb=0, ans=(ll)(1e18);
                    for (int i=1; i<=n; i++) 
                        sumb+=b[i];
                    for (int i=1; i<=n; i++) 
                        ans=min(ans, a[i]+sumb-b[i]);   
                    ret.pb(ans); 
                }
                else {
                    ll sumb=0, ans=(ll)(1e18);
                    for (int i=1; i<=n; i++) 
                        sumb+=b[i];
                    for (int i=1; i<=n; i++) 
                        if (i%2==1)
                            ans=min(ans, a[i]+sumb-b[i]);   
                    ret.pb(ans); 
                }
            }
            return ret; 
        }
    }
    if (q<=5&&A==vector<int>(n, 2)&&B==vector<int>(n, 1)) {
        sort(w+1, w+n+1);
        vector<ll> ret;
        for (int j=1; j<=q; j++) {
            ll ans=0, curr=1; 
            for (int i=1; i<n; i++) {
                if (w[i+1]-w[i]>e[j]) {
                    ans+=curr%2; 
                    curr=1;
                    continue;
                }
                curr++;
            }
        }
        return ret;     
    }
    return vector<ll>(q, -1); 
}

// int32_t main () {
//     ios::sync_with_stdio(false), cin.tie(0);

//     int N_;
//     cin >> N_;
//     vector<int> W(N_), A(N_), B(N_);
//     for (int i=0; i<N_; i++) 
//         cin >> W[i] >> A[i] >> B[i];
//     int Q_;
//     cin >> Q_;
//     vector<int> E(Q_);
//     for (int i=0; i<Q_; i++) 
//         cin >> E[i];
//     vector<ll> ans=calculate_costs(W, A, B, E);
//     cout << "-------------------\n";
//     cout << "Answer:\n"; 
//     for (ll x:ans) {
//         cout << x << ' ';
//     }
//     cout << '\n';
//     return 0;
// }

• Subtask 0: 0 / 0
• Subtask 1: 6 / 6
• Subtask 2: 13 / 13
• Subtask 3: 0 / 17
• Subtask 4: 0 / 11
• Subtask 5: 0 / 20
• Subtask 6: 0 / 15
• Subtask 7: 0 / 18