Submission #1104896

#	Time	Username	Problem	Language	Result	Execution time	Memory
1104896		HiepVu217	Naval battle (CEOI24_battle)	C++17	0 / 100	133 ms	21656 KiB

Main

#include <bits/stdc++.h>
using namespace std;
const int N = 4e5 + 17;
int n, sank[N], xx[N], yy[N];
pair <int, int> ship[N];
char dir[N];
vector <tuple <int, int, int>> vtp, line[N];
void collide (int i, int j)
{
    char d1 = dir[i], d2 = dir[j];
    auto [x1, y1] = ship[i];
    auto [x2, y2] = ship[j];
    int w = (abs (x1 - x2) + abs (y1 - y2)) / 2;
    if (d1 == 'N')
    {
        if (d2 == 'S' && y1 - y2 == w * 2)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'E' && x2 + w == x1 && y1 - w == y2)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'W' && x2 - w == x1 && y1 - w == y2)
        {
            vtp.push_back ({w, i, j});
        }
    }
    if (d1 == 'S')
    {
        if (d2 == 'N' && y2 - y1 == w * 2)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'E' && x2 + w == x1 && y1 + w == y2)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'W' && x2 - w == x1 && y1 + w == y2)
        {
            vtp.push_back ({w, i, j});
        }
    }
    if (d1 == 'E')
    {
        if (d2 == 'W' && x2 - x1 == w * 2)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'N' && x1 + w == x2 && y2 - w == y1)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'S' && x1 + w == x2 && y2 + w == y1)
        {
            vtp.push_back ({w, i, j});
        }
    }
    if (d1 == 'W')
    {
        if (d2 == 'E' && x1 - x2 == w * 2)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'N' && x1 - w == x2 && y2 - w == y1)
        {
            vtp.push_back ({w, i, j});
        }
        if (d2 == 'S' && x1 - w == x2 && y2 + w == y1)
        {
            vtp.push_back ({w, i, j});
        }
    }
}
void battle (int x)
{
    sort (line[x].begin(), line[x].end());
    stack <int> st;
    for (auto [y, id, di]: line[x])
    {
        if (di == 1)
        {
            st.push(id);
            continue;
        }
        if (!st.size())
        {
            continue;
        }
        sank[id] = sank[st.top()] = 1;
        st.pop();
    }
}
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
//    freopen ("battle.inp", "r", stdin);
//    freopen ("battle.out", "w", stdout);
    cin >> n;
    for (int i = 1; i <= n; ++i)
    {
        cin >> xx[i] >> yy[i] >> dir[i];
        ship[i] = {xx[i], yy[i]};
    }
    sort (xx + 1, xx + n + 1);
    sort (yy + 1, yy + n + 1);
    for (int i = 1; i <= n; ++i)
    {
        auto [x, y] = ship[i];
        x = lower_bound (xx + 1, xx + n + 1, x) - xx;
        y = lower_bound (yy + 1, yy + n + 1, y) - yy;
        line[x + y].push_back ({y, i, (dir[i] == 'S' ? 1 : -1)});
    }
    for (int i = 2; i <= 2 * n; i += 2)
    {
        battle(i);
    }
    for (int i = 1; i <= n; ++i)
    {
        if (!sank[i])
        {
            cout << i << "\n";
        }
    }
}

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