Submission #349138

#	Time	Username	Problem	Language	Result	Execution time	Memory
349138		ant101	Wiring (IOI17_wiring)	C++14	100 / 100	65 ms	15604 KiB

wiring

#include "wiring.h"
#include <iostream>
#include <algorithm>
#include <cstring>
#include <iomanip>
#include <fstream>
#include <cmath>
#include <vector>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <map>
#include <stack>
#include <queue>
#include <assert.h>
#include <limits>
#include <cstdio>
#include <complex>
using namespace std;

typedef long long ll;

vector<ll> seqinds;
vector<ll> prefl;
vector<ll> prefr;
vector<ll> dl;
vector<ll> dr;
vector<ll> len;

ll n;
ll ns;

ll INF = 1e18;

void processData(vector<int> &r, vector<int> &b){
    vector<pair<ll, bool>> seq;
    for (ll i : r) {
        seq.push_back({i, false});
    }
    for (ll i : b) {
        seq.push_back({i, true});
    }
    n = seq.size();
    sort(seq.begin(), seq.end());
    ll mx = seq.back().first;
    for (auto p : seq) {
        dl.push_back(p.first);
        dr.push_back(mx-p.first);
    }
    prefl.resize(n+1);
    prefr.resize(n+1);
    for (ll i = 0; i<n; i++){
        prefl[i+1] = prefl[i]+dl[i];
    }
    for (ll i = n-1; i>=0; i--){
        prefr[i] = prefr[i+1]+dr[i];
    }
    
    bool curval = seq[0].second;
    ll curind = 0;
    seqinds.push_back(curind);
    for (ll i = 0; i<n; i++) {
        if (seq[i].second != curval) {
            curind = i;
            seqinds.push_back(curind);
            curval = seq[i].second;
        }
    }
    ns = seqinds.size();
    seqinds.push_back(n);
    for (ll i = 0; i<seqinds.size()-1; i++) {
        len.push_back(seqinds[i+1]-seqinds[i]);
    }
}



void display(string s, vector<ll> &v){
    cout << s << ": ";
    for (ll i : v) {
        cout << i << ' ';
    } cout << '\n';
}

ll lsum(ll i, ll l) {
    ll xi = seqinds[i];
    return prefl[xi+l]-prefl[xi]-l*dl[xi];
}

ll rsum(ll i, ll r) {
    ll xi1 = seqinds[i+1];
    return prefr[xi1-r]-prefr[xi1]-r*dr[xi1];
}

ll performDP() {
    ll mxlen = 0;
    for (ll i : len) {
        mxlen = max(mxlen, i);
    }
    vector<vector<ll>> dp(2, vector<ll>(mxlen+1, INF));
    vector<vector<ll>> dpgeq(2, vector<ll>(mxlen+1, INF));
    bool cur = false;
    dp[!cur][0] = 0;
    dpgeq[!cur][0] = 0;
    for (ll i = 0; i<ns; i++) {
        ll xi1 = seqinds[i+1];
        ll ci = dr[xi1-1]-dr[xi1];
        ll minpref = INF;
        ll maxprev = max(len[i], ((i==0) ? 0 : len[i-1]));
        for (ll r = 0; r<=maxprev; r++) {
            ll newl = len[i]-r;
            if (newl<0) {
                newl = 0;
            }
            ll newmin = dp[!cur][newl]+lsum(i, newl)+rsum(i, len[i]-newl)+ci*newl;
            minpref = min(minpref,newmin);
            dp[cur][r] = minpref+ci*(r-len[i]);
        }
        if (i==ns-1) {
            break;
        }
        for (ll r = maxprev+1; r<=len[i+1]; r++) {
            dp[cur][r] = dp[cur][r-1]+ci;
        }

        for (ll r = max(maxprev, len[i+1])-1; r>=0; r--) {
            dp[cur][r] = min(dp[cur][r], dp[cur][r+1]);
        }
        cur=!cur;
    }
    return dp[cur][0];
}

ll min_total_length(std::vector<int> r, std::vector<int> b) {
    processData(r,b);

    for (ll i = 0; i<r.size(); i++) {
        
    }
    return performDP();
}

Compilation message (stderr)

wiring.cpp: In function 'void processData(std::vector<int>&, std::vector<int>&)':
wiring.cpp:71:21: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   71 |     for (ll i = 0; i<seqinds.size()-1; i++) {
      |                    ~^~~~~~~~~~~~~~~~~
wiring.cpp: In function 'll min_total_length(std::vector<int>, std::vector<int>)':
wiring.cpp:137:21: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  137 |     for (ll i = 0; i<r.size(); i++) {
      |                    ~^~~~~~~~~

• Subtask 1: 7 / 7
• Subtask 2: 13 / 13
• Subtask 3: 10 / 10
• Subtask 4: 25 / 25
• Subtask 5: 45 / 45