제출 #733116

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
733116		t6twotwo	전선 연결 (IOI17_wiring)	C++17	17 / 100	1072 ms	7336 KiB

wiring

#include "wiring.h"
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr ll inf = 1E16;
long long min_total_length(vector<int> R, vector<int> B) {
    int N = R.size();
    int M = B.size();
    vector<array<int, 2>> T;
    for (int x : R) {
        T.push_back({x, 0});
    }
    for (int x : B) {
        T.push_back({x, 1});
    }
    sort(T.begin(), T.end());
    vector<ll> pfs(N + M + 1);
    for (int i = 0; i < N + M; i++) {
        pfs[i + 1] = pfs[i] + T[i][0];
    }
    vector<int> d(N + M, -1);
    vector<ll> dp(N + M + 1, inf);
    dp[0] = 0;
    for (int i = 0; i < N + M; i++) {
        if (i > 0) {
            if (T[i][1] == T[i - 1][1]) {
                d[i] = d[i - 1];
            } else {
                d[i] = i - 1;
            }
        }
        if (d[i] == -1) {
            continue;
        }
        int j = d[i];
        dp[i + 1] = min(dp[i + 1], dp[j + 1] + (pfs[i + 1] - pfs[j + 1]) - 1LL * (i - j) * T[j][0]);
        for (int p = d[j] + 1; p <= j; p++) {
            int m = min(j - p + 1, i - j);
            ll s = (pfs[i + 1] - pfs[i - m + 1]) - (pfs[p + m] - pfs[p]);
            if (m <= j - p) {
                s += 1LL * T[j + 1][0] * ((j + 1) - (p + m)) - (pfs[j + 1] - pfs[p + m]);
            }
            if (m < i - j) {
                s -= 1LL * T[j][0] * ((i - m + 1) - (j + 1)) - (pfs[i - m + 1] - pfs[j + 1]);
            }
            dp[i + 1] = min(dp[i + 1], dp[p] + s);
        }
    }
    return dp[N + M];
}

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