Submission #1044204

#	Time	Username	Problem	Language	Result	Execution time	Memory
1044204		VahanAbraham	Naval battle (CEOI24_battle)	C++17	46 / 100	258 ms	230312 KiB

Main

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <string>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <sstream>
#include <map>
#include <stack>
#include <set>
#include <queue>
#include <list>
#include <unordered_set>
#include <unordered_map>
#include <math.h>
#include <bitset>
#include <cmath>
#include <vector>
#include <iomanip>
#include <random>
#include <chrono>
#include <cassert>

using namespace std;

#define ll long long
#define fr first
#define sc second
#define pb push_back
#define US freopen(".in", "r", stdin); freopen("j.out", "w", stdout);

ll gcd(ll a, ll b)
{
    if (a == 0 || b == 0) {
        return  max(a, b);
    }
    if (a <= b) {
        return gcd(a, b % a);
    }
    else {
        return gcd(a % b, b);
    }
}
ll lcm(ll a, ll b) {
    return (a / gcd(a, b)) * b;
}

const int N = 5005;
const ll oo = 100000000000000000, MOD = 1000000007;


struct NODE {
    int x, y, x1, y1;
    char d;
};

NODE a[N], ner[N];
bool ok[N], lav[N * N];
ll timer[N][N];

void solve() {
    int n;
    cin >> n;
    int mx = 0;
    for (int i = 1; i <= n; ++i) {
        char c;
        cin >> a[i].x >> a[i].y >> a[i].d;
        if (a[i].d == 'N') {
            a[i].x1 = 0;
            a[i].y1 = -1;
        }
        if (a[i].d == 'S') {
            a[i].x1 = 0;
            a[i].y1 = 1;
        }
        if (a[i].d == 'W') {
            a[i].x1 = -1, a[i].y1 = 0;
        }
        if (a[i].d == 'E') {
            a[i].x1 = 1, a[i].y1 = 0;
        }
        ner[i] = a[i];
        mx = max(mx, a[i].x);
        mx = max(mx, a[i].y);
    }
    vector<pair<ll, pair<int, int>>>vec;
    for (int i = 1; i < n; ++i) {
        for (int j = i + 1; j <= n; ++j) {
            if (a[i].d == a[j].d) {
                continue;
            }
            timer[i][j] = oo;
            if (a[i].d == 'S' && a[j].d == 'N') {
                if (a[i].x == a[j].x && a[i].y < a[j].y) {
                    int tm = abs(a[i].y - a[j].y) / 2;
                    timer[i][j] = tm;
                }
            }
            if (a[i].d == 'S' && a[j].d == 'E') {
                if (a[i].y < a[j].y && a[j].x < a[i].x) {
                    int xx = a[i].x, yy = a[j].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'S' && a[j].d == 'W') {
                if (a[i].y < a[j].y && a[j].x > a[i].x) {
                    int xx = a[i].x, yy = a[j].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'N' && a[j].d == 'S') {
                if (a[i].x == a[j].x && a[i].y > a[j].y) {
                    int tm = abs(a[i].y - a[j].y) / 2;
                    timer[i][j] = tm;
                }
            }
            if (a[i].d == 'N' && a[j].d == 'E') {
                if (a[i].y > a[j].y && a[j].x < a[i].x) {
                    int xx = a[i].x, yy = a[j].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'N' && a[j].d == 'W') {
                if (a[i].y > a[j].y && a[j].x > a[i].x) {
                    int xx = a[i].x, yy = a[j].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'W' && a[j].d == 'E') {
                if (a[i].y == a[j].y && a[i].x > a[j].x) {
                    timer[i][j] = abs(a[i].x - a[j].x) / 2;
                }
            }
            if (a[i].d == 'W' && a[j].d == 'S') {
                if (a[i].y > a[j].y && a[i].x > a[j].x) {
                    int xx = a[j].x, yy = a[i].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'W' && a[j].d == 'N') {
                if (a[i].y < a[j].y && a[i].x > a[j].x) {
                    int xx = a[j].x, yy = a[i].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'E' && a[j].d == 'W') {
                if (a[i].y == a[j].y && a[j].x > a[i].x) {
                    timer[i][j] = abs(a[i].x - a[j].x) / 2;
                }
            }
            if (a[i].d == 'E' && a[j].d == 'S') {
                if (a[i].y > a[j].y && a[i].x < a[j].x) {
                    int xx = a[j].x, yy = a[i].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (a[i].d == 'E' && a[j].d == 'N') {
                if (a[i].y < a[j].y && a[i].x < a[j].x) {
                    int xx = a[j].x, yy = a[i].y;
                    ll dist1 = abs(xx - a[i].x) + abs(yy - a[i].y);
                    ll dist2 = abs(xx - a[j].x) + abs(yy - a[j].y);
                    if (dist1 == dist2) {
                        timer[i][j] = dist1;
                    }
                }
            }
            if (timer[i][j] != oo) {
                vec.pb({ timer[i][j],{i,j} });
            }
        }
    }
    sort(vec.begin(), vec.end());
    for (int i = 0; i < vec.size(); ++i){
        int l = i;
        while (l < vec.size() && vec[l].fr == vec[i].fr) {
            if (ok[vec[l].sc.fr] != 1 && ok[vec[l].sc.sc] != 1) {
                lav[l] = 1;
            }
            ++l;
        }
        l = i;
        while (l < vec.size() && vec[l].fr == vec[i].fr) {
            if (lav[l] == 1) {
                ok[vec[l].sc.fr] = 1;
                ok[vec[l].sc.sc] = 1;
            }
            ++l;
        }
        i = l - 1;
    }
    for (int i = 1; i <= n; ++i) {
        if (ok[i] == 0) {
            cout << i << endl;
        }
    }
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    //US
    int tt = 1;
    //cin >> tt;
    while (tt--) {
        solve();
    }
}

Compilation message (stderr)

Main.cpp: In function 'void solve()':
Main.cpp:66:14: warning: unused variable 'c' [-Wunused-variable]
   66 |         char c;
      |              ^
Main.cpp:201:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  201 |     for (int i = 0; i < vec.size(); ++i){
      |                     ~~^~~~~~~~~~~~
Main.cpp:203:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  203 |         while (l < vec.size() && vec[l].fr == vec[i].fr) {
      |                ~~^~~~~~~~~~~~
Main.cpp:210:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, std::pair<int, int> > >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  210 |         while (l < vec.size() && vec[l].fr == vec[i].fr) {
      |                ~~^~~~~~~~~~~~

• 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