제출 #564561

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
564561		RealSnake	Building Skyscrapers (CEOI19_skyscrapers)	C++14	0 / 100	847 ms	16940 KiB

skyscrapers

#include "bits/stdc++.h"
using namespace std;

#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;

#define ll long long
#define mod 1000000007

ofstream fout(".out");
ifstream fin(".in");

map<pair<int, int>, int> mp;
map<pair<int, int>, bool> mp2;
int n, t, j;
int a[1500001];
bool vis[1500001];
int xi[8] = {0, 0, 1, -1, 1, 1, -1, -1};
int yi[8] = {-1, 1, 0, 0, -1, 1, -1, 1};

bool check(int x, int y) {
    return x >= 0 && x < n && y >= 0 && y < n;
}

bool check2(int x, int y) {
    mp2[{x, y}] = 1;
    bool b = 0;
    for(int i = 0; i < 4; i++) {
        int xx = x + xi[i];
        int yy = y + yi[i];
        if(xx >= n || yy >= n)
            b = 1;
        else if(check(xx, yy) && !mp2[{xx, yy}] && (!mp2[{xx, yy}] || mp[{xx, yy}] != n + 1))
            b |= check2(xx, yy);
        if(b)
            return 1;
    }
    return 0;
}

signed main() {

    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> n >> t;
    vector<pair<int, int>> v;
    for(int i = 0; i < n; i++) {
        int x, y;
        cin >> x >> y;
        v.push_back({x, y});
        mp[{x, y}] = i + 1;
    }
    vector<int> vec;
    vec.push_back(0);
    vis[0] = 1;
    int k = 0;
    j = 0;
    int sz = 1;
    while(k < sz) {
        int ind = vec[k++];
        int x = v[ind].first, y = v[ind].second;
        mp2.clear();
        if(!check2(x, y)) {
            if(k == sz)
                break;
            vec.push_back(ind);
            continue;
        }
        mp[{x, y}] = n + 1;
        a[j++] = ind;
        for(int i = 0; i < 8; i++) {
            int xx = x + xi[i];
            int yy = y + yi[i];
            if(check(xx, yy) && mp[{xx, yy}] && mp[{xx, yy}] != n + 1 && !vis[mp[{xx, yy}] - 1]) {
                vis[mp[{xx, yy}] - 1] = 1;
                vec.push_back(mp[{xx, yy}] - 1);
                sz++;
            }
        }
    }
    if(j != n) {
        cout << "NO";
        return 0;
    }
    cout << "YES\n";
    for(int i = 0; i < n; i++)
        cout << a[i] + 1 << "\n";

    return 0;
}

• Subtask 1: 0 / 8
• Subtask 2: 0 / 14
• Subtask 3: 0 / 12
• Subtask 4: 0 / 17
• Subtask 5: 0 / 20
• Subtask 6: 0 / 10
• Subtask 7: 0 / 19