Submission #1134374

#	Time	Username	Problem	Language	Result	Execution time	Memory
1134374		saidpon	Naval battle (CEOI24_battle)	C++20	20 / 100	3094 ms	18756 KiB

Main

#include <bits/stdc++.h>
#define nemeshay ios_base::sync_with_stdio(NULL), cin.tie(0), cout.tie(0);
#define int long long
#define sigma signed
#define pb push_back
#define fr first
#define sc second
#define pii pair<int, int>
using namespace std;
const int N = 2e5 + 2, inf = 1e18 + 7;
pair<pii, char> a[N];
int used[N];
sigma main() {
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i].fr.fr >> a[i].fr.sc >> a[i].sc;
    if (n == 2) {
        if (a[1].sc == a[2].sc) {
            cout << "1\n2";
            return 0;
        }
        if (a[1].sc == 'S' && a[2].sc == 'N') return 0;
        if (a[1].sc == 'N' && a[2].sc == 'S') return 0;
        if (a[1].sc == 'E' && a[2].sc == 'W') return 0;
        if (a[1].sc == 'W' && a[2].sc == 'E') return 0;
        if (a[1].sc == 'S') {
            if (a[2].sc == 'E') {
                if (a[1].fr.sc > a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr > a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
            else {
                if (a[1].fr.sc > a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr < a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
        }
        else if (a[1].sc == 'N') {
            if (a[2].sc == 'E') {
                if (a[1].fr.sc < a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr > a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
            else {
                if (a[1].fr.sc < a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr < a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
        }
        else if (a[1].sc == 'E') {
            swap(a[1], a[2]);
            if (a[1].sc == 'N') {
                if (a[1].fr.sc < a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr > a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
            else {
                if (a[1].fr.sc > a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr > a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
        }
        else {
            swap(a[1], a[2]);
            if (a[1].sc == 'N') {
                if (a[1].fr.sc < a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr < a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
            else {
                if (a[1].fr.sc > a[2].fr.sc) {
                    cout << "1\n2";
                    return 0;
                }
                if (a[2].fr.fr < a[1].fr.fr) {
                    cout << "1\n2";
                    return 0;
                }
                if (abs(a[1].fr.sc - a[2].fr.sc) == abs(a[1].fr.fr - a[2].fr.fr)) return 0;
                cout << "1\n2";
            }
        }
        return 0;
    }
    int q = N;
    while (q--) {
        map <pii, int> pon;
        for (int i = 1; i <= n; i++) {
            if (used[i]) continue;
            if (a[i].sc == 'N') a[i].fr.sc--;
            else if (a[i].sc == 'S') a[i].fr.sc++;
            else if (a[i].sc == 'W') a[i].fr.fr--;
            else a[i].fr.fr++;
            int x = a[i].fr.fr, y = a[i].fr.sc;
            if (pon[{x, y}] == 0) pon[{x, y}] = i;
            else used[i] = 1, used[pon[{x, y}]] = 1;
        }
    }
    for (int i = 1; i <= n; i++) if (!used[i]) cout << i << endl;
}

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