Submission #420188

#	Time	Username	Problem	Language	Result	Execution time	Memory
420188		temurbek_khujaev	Wiring (IOI17_wiring)	C++17	0 / 100	1 ms	204 KiB

wiring

#include <bits/stdc++.h>
#include "wiring.h"

using namespace std;
#define N 402
long long dp[N];
const long long INF = 1e18;

long long min_total_length(std::vector<int> r, std::vector<int> b) {
    vector<pair<int, int >> v;
    for (auto x:r) v.push_back({x, -1});
    for (auto x:b) v.push_back({x, 1});
    v.push_back({-1, -2});
    sort(v.begin(), v.end());

    dp[0] = 0;
    for (int i = 1; i < v.size(); i++) {
        dp[i] = INF;
        int balance = 0;
        for (int j = i; j > 0; j--) {
            balance += v[j].second;
            bool flag = true;
            if (v[i].second != v[j].second && flag) {
                dp[i] = min(dp[i], dp[j - 1] + v[i].first - v[j].first);
                flag = false;
            }
            if (balance == 0) {
                long long cost = 0;
                vector<int> st[3];
                for (int k = j; k <= i; k++) {
                    st[v[k].second + 1].push_back(v[k].first);
                    if (!st[0].empty() && !st[2].empty()) {
                        cost += abs(st[2].back() - st[0].back());
                        st[0].pop_back();
                        st[1].pop_back();
                    }
                }
                dp[i] = min(dp[i], dp[j - 1] + cost);
            }
        }
//        cerr << dp[i] << ' ';
    }
    return dp[v.size() - 1];
}

Compilation message (stderr)

wiring.cpp: In function 'long long int min_total_length(std::vector<int>, std::vector<int>)':
wiring.cpp:17:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   17 |     for (int i = 1; i < v.size(); i++) {
      |                     ~~^~~~~~~~~~

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