Submission #1026364

#	Time	Username	Problem	Language	Result	Execution time	Memory
1026364		12345678	Naval battle (CEOI24_battle)	C++17	6 / 100	1 ms	600 KiB

Main

#include <bits/stdc++.h>

using namespace std;

const int nx=5e3+5;

int n, x[nx], y[nx], dn[nx];
char dr[nx];
map<int, map<pair<int, int>, set<int>>> mp;

tuple<int, int, int> crash(int i, int j)
{
    if (dr[i]==dr[j]) return {0, 0, 0};
    if ((dr[i]=='N'&&dr[j]=='S')||(dr[i]=='S'&&dr[j]=='N')) 
    {
        if (x[i]==x[j]&&(abs(y[i]-y[j])%2)==0) return {abs(y[i]-y[j])/2, x[i], min(y[i], y[j])+abs(y[i]-y[j])/2};
        else return {0, 0, 0};
    }
    if ((dr[i]=='W'&&dr[j]=='E')||(dr[i]=='E'&&dr[j]=='W')) 
    {
        if (y[i]==y[j]&&(abs(x[i]-x[j])%2)==0) return {abs(x[i]-x[j])/2, min(x[i], x[j])+abs(x[i]-x[j])/2, y[i]};
        else return {0, 0, 0};
    }
    if (dr[i]=='N')
    {
        if (dr[j]=='W'&&y[i]>y[j]&&y[i]-y[j]==x[j]-x[i]) return {y[i]-y[j], x[i], y[j]};
        if (dr[j]=='E'&&y[i]>y[j]&&y[i]-y[j]==x[i]-x[j]) return {y[i]-y[j], x[i], y[j]};
    }
    if (dr[i]=='S')
    {
        if (dr[j]=='W'&&y[i]<y[j]&&y[j]-y[i]==x[j]-x[i]) return {y[j]-y[i], x[i], y[j]};
        if (dr[j]=='E'&&y[i]<y[j]&&y[j]-y[i]==x[i]-x[j]) return {y[j]-y[i], x[i], y[j]};
    }
    if (dr[j]=='N')
    {
        if (dr[i]=='W'&&y[j]>y[i]&&y[j]-y[i]==x[i]-x[j]) return {y[j]-y[i], x[j], y[i]};
        if (dr[i]=='E'&&y[j]>y[i]&&y[j]-y[i]==x[j]-x[i]) return {y[j]-y[i], x[j], y[i]};
    }
    if (dr[j]=='S')
    {
        if (dr[i]=='W'&&y[j]<y[i]&&y[i]-y[j]==x[i]-x[j]) return {y[i]-y[j], x[j], y[i]};
        if (dr[i]=='E'&&y[j]<y[i]&&y[i]-y[j]==x[j]-x[i]) return {y[i]-y[j], x[j], y[i]};
    }
    return {0, 0, 0};
}

int main()
{
    cin.tie(NULL)->sync_with_stdio(false);
    cin>>n;
    for (int i=1; i<=n; i++) cin>>x[i]>>y[i]>>dr[i];
    for (int i=1; i<=n; i++) 
    {
        for (int j=i+1; j<=n; j++)
        {
            auto [x, y, z]=crash(i, j);
            //cout<<"debug "<<i<<' '<<j<<' '<<x<<' '<<y<<' '<<z<<'\n';
            if (x==0) continue;
            mp[x][{y, z}].insert(i);
            mp[x][{y, z}].insert(j);
        }
    }
    for (auto [t, x]:mp)
    {
        for (auto [c, s]:x)
        {
            int cnt=0;
            for (auto p:s) if (!dn[p]) cnt++;
            if (cnt>1||1) for (auto p:s) dn[p]=1;
        }
    }
    for (int i=1; i<=n; i++) if (!dn[i]) cout<<i<<'\n';
}

• Subtask 0: 0 / 0
• Subtask 1: 6 / 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