제출 #1062338

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

wiring

// 人外有人，天外有天
// author: Ausp3x

#pragma GCC optimize("O1, O2, O3, Ofast, unroll-loops")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include "wiring.h"
using namespace std;
using namespace __gnu_pbds;

#define fi first
#define se second
#define pb push_back
// #define DEBUG
typedef long long    lng;
template<class T> 
using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;

int const INF32 = 0x3f3f3f3f;
lng const INF64 = 0x3f3f3f3f3f3f3f3f;

vector<int> st1_R, st1_B;
vector<vector<lng>> memo;
lng st1_dp(int i, int j) {
    if (i < 0 || j < 0)
        return INF64;

    if (memo[i][j] != INF64)
        return memo[i][j];
    
    if (i == 0 && j == 0)
        return memo[i][j] = abs(st1_R[i] - st1_B[j]);

    return memo[i][j] = min(min({st1_dp(i - 1, j), st1_dp(i, j - 1), st1_dp(i - 1, j - 1)}) + abs(st1_R[i] - st1_B[j]), INF64);
}

lng min_total_length(vector<int> R, vector<int> B) {
    int n = R.size(), m = B.size();
    

    if (n <= 200 && m <= 200) {
        st1_R = R;
        st1_B = B;
        memo.resize(n, vector<lng>(m, INF64));

        #ifdef DEBUG
        cout << st1_dp(n - 1, m - 1) << endl;
        #endif DEBUG
        #ifndef DEBUG
        return st1_dp(n - 1, m - 1);
        #endif
    }

    if (R.back() < B[0]) {
        int i = n - 1, j = m - 1;
        lng ans = 0;
        while (i >= 0 && j >= 0) {
            ans += B[j] - R[i];
            
            i--;
            j--;
        }

        while (i >= 0) {
            ans += B[0] - R[i];
            i--;
        }

        while (j >= 0) {
            ans += B[j] - R[n - 1];
            j--;
        }

        return ans;
    }

    return -1;
}

#ifdef DEBUG
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int t = 1;
    cin >> t;
    while (t--) {
        int n, m;
        cin >> n >> m;
        vector<int> R(n);
        for (int &r : R)
            cin >> r;
        vector<int> B(m);
        for (int &b : B)
            cin >> b;

        cout << min_total_length(R, B) << endl;
    }

    return 0;
}
#endif

컴파일 시 표준 에러 (stderr) 메시지

wiring.cpp:48:16: warning: extra tokens at end of #endif directive [-Wendif-labels]
   48 |         #endif DEBUG
      |                ^~~~~
wiring.cpp:4:55: warning: bad option '-f O2' to pragma 'optimize' [-Wpragmas]
    4 | #pragma GCC optimize("O1, O2, O3, Ofast, unroll-loops")
      |                                                       ^
wiring.cpp:4:55: warning: bad option '-f O3' to pragma 'optimize' [-Wpragmas]
wiring.cpp:4:55: warning: bad option '-f Ofast' to pragma 'optimize' [-Wpragmas]
wiring.cpp:4:55: warning: bad option '-f unroll-loops' to pragma 'optimize' [-Wpragmas]
In file included from wiring.cpp:7:
wiring.h:3:66: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
    3 | long long min_total_length(std::vector<int> r, std::vector<int> b);
      |                                                                  ^
wiring.h:3:66: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
wiring.h:3:66: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
wiring.h:3:66: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
wiring.h:3:66: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
wiring.h:3:66: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
wiring.h:3:66: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
wiring.h:3:66: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
wiring.cpp:24:24: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
   24 | lng st1_dp(int i, int j) {
      |                        ^
wiring.cpp:24:24: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
wiring.cpp:24:24: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
wiring.cpp:24:24: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
wiring.cpp:37:50: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
   37 | lng min_total_length(vector<int> R, vector<int> B) {
      |                                                  ^
wiring.cpp:37:50: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
wiring.cpp:37:50: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
wiring.cpp:37:50: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]

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