Submission #1329315

#TimeUsernameProblemLanguageResultExecution timeMemory
1329315DanielPr8Wiring (IOI17_wiring)C++20
17 / 100
1093 ms10772 KiB
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using vll = vector<ll>;
using vvl = vector<vll>;
using pll = pair<ll,ll>;
using vpl = vector<pll>;
using vvp = vector<vpl>;
#define f first
#define s second
#define pb push_back
#define all(v) v.begin(),v.end()
vpl ord;ll n;
vll rs;
ll ran(ll l, ll r, ll m){
    if(r-m>m-l){
        return -ord[m-1].f*(r-m-m+l);
    }
    else{
       return ord[m].f*(m-l-r+m);
    }
}
long long min_total_length(std::vector<int> r, std::vector<int> b) {
    ord.clear();
    for(ll i: r)ord.pb({i,1});
    for(ll i: b)ord.pb({i,0});
    sort(all(ord));
    n = ord.size();
    rs = vll(n+1);
    vll st={0};
    for(ll i = 0; i < n; ++i){
        rs[i+1]=rs[i]+ord[i].f;
        if(i+1<n && ord[i+1].s!=ord[i].s){st.pb(i+1);}
    }
    st.pb(n);
    reverse(all(st));
    vll dp(n+1, 1e17);
    vll dp1(n+1, 1e17);
    ll las=0, bf=0;
    dp[0]=0;
    dp1[0]=-rs[st.back()];
    while(st.size()>1){
        bf=las;las=st.back();st.pop_back();
        
        for(ll i = las; i < st.back(); ++i){
            ll cur = 1e17;
            for(ll j = bf+1; j < las && j<2*las-i; ++j){
                cur = min(cur, dp1[j]+ord[las].f*(2*las-j-i-1));
            }
            dp[i+1]=min(dp[i+1],cur);
        }
        for(ll i = st.back()-1; i >= las; --i){
            ll cur = 1e17;
            for(ll j = max(bf+1,2*las-i); j < las; ++j){
                cur = min(cur, dp1[j]-ord[las-1].f*(i+1-2*las+j));
            }
            dp[i+1]=min(dp[i+1], cur);
        }

        
        for(ll i = st.back()-1; i >= las; --i){
            dp[i+1]+=rs[i+1]-rs[las];
            if(bf!=las)dp[i+1]=min(dp[i+1], dp[bf]+ran(bf,i+1,las)+rs[i+1]-rs[las]-rs[las]+rs[bf]);
            if(bf+1==las)dp[i+1]=min(dp[i+1], dp[bf+1]+ran(bf,i+1,las)+rs[i+1]-rs[las]-rs[las]+rs[bf]);
            dp1[i+1]=dp[i+1];
            dp1[i+1]+=-rs[st.back()]+rs[i+1];
        }
    }
    return dp[n];
}

ll ran2(ll l, ll r, ll m){
    ll ans=0;
    if(r-m>m-l){
        for(ll i = m; i < r;++i){
            ans += abs(ord[i].f-ord[min(m-1,i-m+l)].f);
        }
    }
    else{
        for(ll i = m-1; i >=l;--i){
            ans += abs(ord[i].f-ord[max(m,i-m+r)].f);
        }
    }
    return ans;
}

long long min_total_length2(std::vector<int> r, std::vector<int> b) {
    ord.clear();
    for(ll i: r)ord.pb({i,1});
    for(ll i: b)ord.pb({i,0});
    sort(all(ord));
    n = ord.size();
    vll dp(n+1, 1e17);
    ll las=0, bf=0;
    dp[0]=0;
    for(ll i = 1; i < n; ++i){
        if(ord[i].s!=ord[i-1].s){bf=las;las=i;}
        if(las==bf+1){
            dp[i+1]=min(dp[i+1], min(dp[bf],dp[bf+1])+ran2(bf,i+1,las));
        }
        else{
            for(ll j = bf; j < las; ++j){
                dp[i+1] = min(dp[i+1], dp[j]+ran2(j,i+1, las));
            }
        }
    }
    return dp[n];
}

// int main(){
//     ll n=3,m=2,q=1000;
//     while(q--){
//         vll qu(n+m);
//         for(ll i = 1; i < n+m; ++i)qu[i]=qu[i-1]+rand()%3+1;
//         random_shuffle(all(qu));
//         vector<int> a,b;
//         for(ll i = 0; i < n; ++i)a.pb(qu[i]);
//         for(ll i = n; i < n+m; ++i)b.pb(qu[i]);
//         if(min_total_length(a,b)!=min_total_length2(a,b)){
//             for(ll i:qu)cout << i << " ";
//             cout << "\n";
//             cout << min_total_length(a,b);
//             return 0;
//         }
//     }
// }


// int main() {
// 	int n, m;
//     assert(2 == scanf("%d %d", &n, &m));

// 	vector<int> r(n), b(m);
// 	for(int i = 0; i < n; i++)
// 		assert(1 == scanf("%d", &r[i]));
// 	for(int i = 0; i < m; i++)
// 		assert(1 == scanf("%d", &b[i]));

// 	long long res = min_total_length(r, b);
// 	printf("%lld\n", res);

// 	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...