Submission #1103243

#TimeUsernameProblemLanguageResultExecution timeMemory
1103243koukirocksNaval battle (CEOI24_battle)C++17
6 / 100
1998 ms1048576 KiB
#include <bits/stdc++.h>
#define speed ios_base::sync_with_stdio(0); cin.tie(0)
#define all(x) (x).begin(),(x).end()
#define F first
#define S second
//#pragma GCC optimize("O3,unroll-loops")
//#pragma GCC target("avx,avx2")
//#pragma GCC target("popcnt")
 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef double db;
typedef long double ldb;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
 
const ll MAX=2e5+10,P=1e9+7;
const ll INF=0x3f3f3f3f,oo=0x3f3f3f3f3f3f3f3f;
const ldb eps=1e-6;
const ldb PI=acos(-1.0);
const int dir[4][2]={{0,-1},{0,1},{1,0},{-1,0}};
template<typename T>
using vvector = vector<vector<T>>;

int main() {
    speed;
    int n;
    cin>>n;
    vector<pair<pll,int>> p(n+1);
    for (int i=1;i<=n;i++) {
        char c;
        cin>>p[i].F.F>>p[i].F.S>>c;
        if (c=='N') p[i].S=0;
        else if (c=='S') p[i].S=1;
        else if (c=='E') p[i].S=2;
        else p[i].S=3;
    }
    vector<pair<int,pll>> coll;
    for (int i=1;i<=n;i++) {
        for (int j=i+1;j<=n;j++) {
            if (p[i].S==p[j].S) continue;
            if (dir[p[i].S][0]==0 and dir[p[j].S][0]==0) {
                if (p[i].F.F!=p[j].F.F) continue;
                ll d = abs(p[i].F.S-p[j].F.S);
                coll.push_back(pair(d/2,pair(i,j)));
            } else if (dir[p[i].S][1]==0 and dir[p[j].S][1]==0) {
                if (p[i].F.S!=p[j].F.S) continue;
                ll d = abs(p[i].F.F-p[j].F.F);
                coll.push_back(pair(d/2,pair(i,j)));
            } else if (dir[p[i].S][0]==0) {
                int t1=(p[j].F.S-p[i].F.S)/dir[p[i].S][1];
                int t2=(p[i].F.F-p[j].F.F)/dir[p[j].S][0];
                if (t1<0 or t2<0 or t1!=t2) continue;
                coll.push_back(pair(t1,pair(i,j)));
            } else {
                int t1=(p[j].F.F-p[i].F.F)/dir[p[i].S][0];
                int t2=(p[i].F.S-p[j].F.S)/dir[p[j].S][1];
                if (t1<0 or t2<0 or t1!=t2) continue;
                coll.push_back(pair(t1,pair(i,j)));
            }
        }
    }
    vector<ll> ans(n+1,oo);
    sort(all(coll));
    reverse(all(coll));
    while (!coll.empty()) {
        auto [t,ij] = coll.back();
        if (ans[ij.F]>=t and ans[ij.S]>=t) {
            ans[ij.F]=t;
            ans[ij.S]=t;
        }
        coll.pop_back();
    }
    for (int i=1;i<=n;i++) {
        if (ans[i]==oo) cout<<i<<"\n";
    }
    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...