제출 #1329252

#제출 시각아이디문제언어결과실행 시간메모리
1329252DanielPr8Wiring (IOI17_wiring)C++20
17 / 100
1095 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;
    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()];
    for(ll i = 1; i < n; ++i){
        if(st.back()==i){bf=las;las=i;st.pop_back();}
        for(ll j = bf+1; j < las && j<2*las+i; ++j){
            dp[i+1] = min(dp[i+1], dp1[j]+ran(j,i+1, las));
        }
        for(ll j = max(bf+1,2*las+i); j < las; ++j){
            dp[i+1] = min(dp[i+1], dp1[j]+ran(j,i+1, las));
        }
        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=4,m=3,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...