Submission #1074690

#	Time	Username	Problem	Language	Result	Execution time	Memory
1074690		Ausp3x	Naval battle (CEOI24_battle)	C++17	46 / 100	2798 ms	1048576 KiB

Main

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

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

#define fi first
#define se second
#define pb push_back
// #define DEBUG
typedef long long    lng;
typedef __int128     lli;
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;

map<char, pair<int, int>> d_to_move = {
    {'N', {0, -1}},
    {'S', {0, 1}},
    {'E', {1, 0}},
    {'W', {-1, 0}}
};

lng getCollisionTime(tuple<int, int, char> &ship1, tuple<int, int, char> &ship2) {
    auto [x1, y1, d1] = ship1;
    auto [x2, y2, d2] = ship2;
    
    if (d1 == d2)
        return INF64;
    
    if (d1 == 'N') {
        if (d2 == 'S') {
            if (x1 == x2 && y1 >= y2 && abs(y1 - y2) % 2 == 0) 
                return abs(y1 - y2) / 2;
            else
                return INF64;
        } else if (d2 == 'E') {
            if (x1 >= x2 && y1 >= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        } else if (d2 == 'W') {
            if (x1 <= x2 && y1 >= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        }
    } else if (d1 == 'S') {
        if (d2 == 'N') {
            if (x1 == x2 && y1 <= y2 && abs(y1 - y2) % 2 == 0) 
                return abs(y1 - y2) / 2;
            else
                return INF64;
        } else if (d2 == 'E') {
            if (x1 >= x2 && y1 <= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        } else if (d2 == 'W') {
            if (x1 <= x2 && y1 <= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        }
    } else if (d1 == 'E') {
        if (d2 == 'N') {
            if (x1 <= x2 && y1 <= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        } else if (d2 == 'S') {
            if (x1 <= x2 && y1 >= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        } else if (d2 == 'W') {
            if (x1 <= x2 && y1 == y2 && abs(x1 - x2) % 2 == 0) 
                return abs(x1 - x2) / 2;
            else
                return INF64;
        }
    } else if (d1 == 'W') {
        if (d2 == 'N') {
            if (x1 >= x2 && y1 <= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        } else if (d2 == 'S') {
            if (x1 >= x2 && y1 >= y2 && abs(x1 - x2) == abs(y1 - y2))
                return abs(x1 - x2);
            else
                return INF64;
        } else if (d2 == 'E') {
            if (x1 >= x2 && y1 == y2 && abs(x1 - x2) % 2 == 0) 
                return abs(x1 - x2) / 2;
            else
                return INF64;
        }
    }

    return -1;
}

void solve() {
    int n;
    cin >> n;
    vector<tuple<int, int, char>> ships(n + 1);
    for (int i = 1; i <= n; i++)
        cin >> get<0>(ships[i]) >> get<1>(ships[i]) >> get<2>(ships[i]);
    
    priority_queue<tuple<lng, int, int>, vector<tuple<lng, int, int>>, greater<tuple<lng, int, int>>> pq; 
    for (int i = 1; i <= n; i++)
        for (int j = i + 1; j <= n; j++) {
            lng t = getCollisionTime(ships[i], ships[j]);

            assert(t != -1);

            if (t == INF64) 
                continue;

            // cout << t << ' ' << i << ' ' << j << endl;
            pq.push({t, i, j});
        }

    vector<int> death_time(n + 1, -1);
    while (!pq.empty()) {
        auto [t, i, j] = pq.top();
        pq.pop();

        if ((death_time[i] != -1 && death_time[i] != t) || (death_time[j] != -1 && death_time[j] != t))
            continue;

        death_time[i] = t;
        death_time[j] = t;
    }

    // for (int x : death_time)
    //     cout << x << ' ';
    // cout << endl;

    // cout << "ans: ";
    for (int i = 1; i <= n; i++)
        if (death_time[i] == -1)
            cout << i << ' ';
    cout << endl;

    return;
}

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

    int t = 1;
    // cin >> t;
    while (t--) {
        solve();
    }

    return 0;
}

Compilation message (stderr)

Main.cpp:4:55: warning: bad option '-f O2' to pragma 'optimize' [-Wpragmas]
    4 | #pragma GCC optimize("O1, O2, O3, Ofast, unroll-loops")
      |                                                       ^
Main.cpp:4:55: warning: bad option '-f O3' to pragma 'optimize' [-Wpragmas]
Main.cpp:4:55: warning: bad option '-f Ofast' to pragma 'optimize' [-Wpragmas]
Main.cpp:4:55: warning: bad option '-f unroll-loops' to pragma 'optimize' [-Wpragmas]
Main.cpp:29:80: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
   29 | lng getCollisionTime(tuple<int, int, char> &ship1, tuple<int, int, char> &ship2) {
      |                                                                                ^
Main.cpp:29:80: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
Main.cpp:29:80: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
Main.cpp:29:80: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
Main.cpp:109:12: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
  109 | void solve() {
      |            ^
Main.cpp:109:12: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
Main.cpp:109:12: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
Main.cpp:109:12: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
Main.cpp:155:10: warning: bad option '-f O2' to attribute 'optimize' [-Wattributes]
  155 | int main() {
      |          ^
Main.cpp:155:10: warning: bad option '-f O3' to attribute 'optimize' [-Wattributes]
Main.cpp:155:10: warning: bad option '-f Ofast' to attribute 'optimize' [-Wattributes]
Main.cpp:155:10: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]

• Subtask 0: 0 / 0
• Subtask 1: 6 / 6
• Subtask 2: 12 / 12
• Subtask 3: 8 / 8
• Subtask 4: 11 / 11
• Subtask 5: 9 / 9
• Subtask 6: 0 / 30
• Subtask 7: 0 / 24