oj.uz

답안 #1095968

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1095968 2024-10-03T13:45:54 Z yoav_s Building Skyscrapers (CEOI19_skyscrapers) C++17
22 / 100
 3500 ms 20292 KB 
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef vector<ll> v;
typedef vector<v> vv;
typedef vector<vv> vvv;
typedef pair<ll, ll> p;
typedef vector<p> vp;
typedef vector<vp> vvp;
typedef vector<vvp> vvvp;
typedef pair<ll, p> tri;
typedef vector<tri> vtri;
typedef vector<vtri> vvtri;
typedef vector<vvtri> vvvtri;
typedef vector<bool> vb;
typedef vector<vb> vvb;
typedef vector<vvb> vvvb;

#define f first
#define s second
#define pb push_back
#define eb emplace_back
#define all(v) (v).begin(),(v).end()

const ll INF = 1e18;
const ll mod = 1e9 + 7;

p getParent(p cur, vvp &dsuParent)
{
    if (dsuParent[cur.f][cur.s] == cur) return cur;
    return dsuParent[cur.f][cur.s] = getParent(dsuParent[cur.f][cur.s], dsuParent);
}

int main()
{
    ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
    ll n, t;
    cin >> n >> t;
    vp cells(n);
    for (ll i=  0; i < n; i++) cin >> cells[i].f >> cells[i].s;
    v relevantX, relevantY;
    for (ll i = 0; i < n; i++)
    {
        relevantX.pb(cells[i].f - 1);
        relevantX.pb(cells[i].f);
        relevantX.pb(cells[i].f + 1);
        relevantY.pb(cells[i].s - 1);
        relevantY.pb(cells[i].s);
        relevantY.pb(cells[i].s + 1);
    }
    sort(all(relevantX)); sort(all(relevantY));
    relevantX.erase(unique(all(relevantX)), relevantX.end()); relevantY.erase(unique(all(relevantY)), relevantY.end());
    map<ll, ll> xShrink;
    map<ll, ll> yShrink;
    for (ll i = 0; i < relevantX.size(); i++) xShrink[relevantX[i]] = i + 1;
    for (ll i = 0; i < relevantY.size(); i++) yShrink[relevantY[i]] = i + 1;
    for (ll i = 0; i < n; i++) cells[i] = p(xShrink[cells[i].f], yShrink[cells[i].s]);
    ll maxX = relevantX.size() + 2, maxY = relevantY.size() + 2;
    vvb occupied(maxX, vb(maxY, false));
    for (ll i = 0; i < n; i++) occupied[cells[i].f][cells[i].s] = true;
    set<ll, greater<ll>> available;
    for (ll i = 0; i < n; i++) available.insert(i);
    v order;
    vb possible(n, false);
    vv adjacentStructures(n);
    for (ll i = 0; i < n; i++)
    {
        for (ll j = i + 1; j < n; j++)
        {
            if (abs(cells[i].f - cells[j].f) <= 1 && abs(cells[i].s - cells[j].s) <= 1)
            {
                adjacentStructures[i].pb(j);
                adjacentStructures[j].pb(i);
            }
        }
    }
    vb visited(n, false);
    stack<ll> dfs; dfs.push(0);
    ll cnt = 0;
    while (!dfs.empty())
    {
        auto top = dfs.top();
        dfs.pop();
        if (visited[top]) continue;
        visited[top] = true; cnt++;
        for (ll x : adjacentStructures[top]) dfs.push(x);
    }
    if (cnt != n)
    {
        cout << "NO\n";
        return 0;
    }
    vvp dsuParent(maxX, vp(maxY));
    for (ll i = 0; i < maxX; i++) for (ll j = 0; j < maxY; j++) dsuParent[i][j] = p(i, j);
    vv dsuSize(maxX, v(maxY, 1));
    for (ll i = 0; i < maxX; i++)
    {
        for (ll j = 0; j < maxY; j++)
        {
            if (occupied[i][j]) continue;
            vp neighbors;
            if (i - 1 >= 0)
            {
                neighbors.eb(i - 1, j);
            }
            if (i + 1 < maxX)
            {
                neighbors.eb(i + 1, j);
            }
            if (j - 1 >= 0)
            {
                neighbors.eb(i, j - 1);
            }
            if (j + 1 < maxY)
            {
                neighbors.eb(i, j + 1);
            }
            for (auto x : neighbors)
            {
                if (occupied[x.f][x.s]) continue;
                p p1 = getParent(p(i, j), dsuParent), p2 = getParent(x, dsuParent);
                if (p1 == p2) continue;
                if (dsuSize[p1.f][p1.s] > dsuSize[p2.f][p2.s]) swap(p1,p2);
                dsuParent[p1.f][p1.s] = p2;
                dsuSize[p2.f][p2.s] += dsuSize[p1.f][p1.s];
            }
        }
    }
    for (ll i = n - 1; i >= 0; i--)
    {
        bool done = false;
        for (ll x : available)
        {
            vp neighbors;
            if (cells[x].f - 1 >= 0)
            {
                neighbors.eb(cells[x].f - 1, cells[x].s);
            }
            if (cells[x].f + 1 < maxX)
            {
                neighbors.eb(cells[x].f + 1, cells[x].s);
            }
            if (cells[x].s - 1 >= 0)
            {
                neighbors.eb(cells[x].f, cells[x].s - 1);
            }
            if (cells[x].s + 1 < maxY)
            {
                neighbors.eb(cells[x].f, cells[x].s + 1);
            }
            bool hasNeighbor = false;
            p expected = getParent(p(0, 0), dsuParent);
            for (auto y : neighbors)
            {
                if (!occupied[y.f][y.s] && getParent(y, dsuParent) == expected) hasNeighbor = true;
            }
            bool poss = true;
            if (!hasNeighbor) poss = false;
            else if (i > 1)
            {
                vb visited(n, false);
                auto ptr = available.begin();
                if (*ptr == x) ptr++;
                ll start = *ptr;
                stack<ll> dfs; dfs.push(start);
                ll cnt = 0;
                while (!dfs.empty())
                {
                    auto top = dfs.top();
                    dfs.pop();
                    if (available.count(top) == 0 || top == x || visited[top]) continue;
                    visited[top] = true; cnt++;
                    for (ll x : adjacentStructures[top]) dfs.push(x);
                }
                if (cnt != available.size() - 1)
                {
                    poss = false;
                }
            }
            if (!poss)
            {
                occupied[cells[x].f][cells[x].s] = true;
                continue;
            }
            for (auto y : neighbors)
            {
                if (!occupied[y.f][y.s])
                {
                    p p1 = getParent(y, dsuParent), p2 = getParent(cells[x], dsuParent);
                    if (p1 == p2) continue;
                    if (dsuSize[p1.f][p1.s] > dsuSize[p2.f][p2.s]) swap(p1,p2);
                    dsuParent[p1.f][p1.s] = p2;
                    dsuSize[p2.f][p2.s] += dsuSize[p1.f][p1.s];
                }
            }
            occupied[cells[x].f][cells[x].s] = false;
            available.erase(x);
            order.pb(x);
            done = true;
            break;
        }
        assert(done);
    }
    reverse(all(order));
    cout << "YES\n";
    for (ll x : order) cout << x + 1 << "\n";
    return 0;
}

Compilation message

skyscrapers.cpp: In function 'int main()':
skyscrapers.cpp:57:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   57 |     for (ll i = 0; i < relevantX.size(); i++) xShrink[relevantX[i]] = i + 1;
      |                    ~~^~~~~~~~~~~~~~~~~~
skyscrapers.cpp:58:22: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   58 |     for (ll i = 0; i < relevantY.size(); i++) yShrink[relevantY[i]] = i + 1;
      |                    ~~^~~~~~~~~~~~~~~~~~
skyscrapers.cpp:177:25: warning: comparison of integer expressions of different signedness: 'll' {aka 'long long int'} and 'std::set<long long int, std::greater<long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  177 |                 if (cnt != available.size() - 1)
      |                     ~~~~^~~~~~~~~~~~~~~~~~~~~~~

Subtask #1
8.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 0 ms 348 KB ans=NO N=4
4 Correct 0 ms 348 KB ans=YES N=5
5 Correct 0 ms 348 KB ans=YES N=9
6 Correct 0 ms 348 KB ans=YES N=5
7 Correct 0 ms 348 KB ans=NO N=9
8 Correct 0 ms 348 KB ans=NO N=10
9 Correct 0 ms 348 KB ans=YES N=10
10 Correct 0 ms 348 KB ans=YES N=10
11 Correct 0 ms 348 KB ans=YES N=10
12 Correct 0 ms 348 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 0 ms 348 KB ans=YES N=8
15 Correct 1 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 0 ms 348 KB ans=NO N=4
4 Correct 0 ms 348 KB ans=YES N=5
5 Correct 0 ms 348 KB ans=YES N=9
6 Correct 0 ms 348 KB ans=YES N=5
7 Correct 0 ms 348 KB ans=NO N=9
8 Correct 0 ms 348 KB ans=NO N=10
9 Correct 0 ms 348 KB ans=YES N=10
10 Correct 0 ms 348 KB ans=YES N=10
11 Correct 0 ms 348 KB ans=YES N=10
12 Correct 0 ms 348 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 0 ms 348 KB ans=YES N=8
15 Correct 1 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
17 Correct 0 ms 348 KB ans=YES N=17
18 Correct 0 ms 348 KB ans=YES N=25
19 Correct 2 ms 348 KB ans=YES N=100
20 Correct 9 ms 344 KB ans=YES N=185
21 Correct 1 ms 348 KB ans=NO N=174
22 Correct 1 ms 348 KB ans=YES N=90
23 Correct 1 ms 344 KB ans=YES N=63
24 Correct 1 ms 348 KB ans=YES N=87
25 Correct 5 ms 348 KB ans=YES N=183
26 Correct 7 ms 348 KB ans=YES N=188
27 Correct 10 ms 348 KB ans=YES N=183
28 Correct 6 ms 348 KB ans=YES N=189
29 Correct 9 ms 348 KB ans=YES N=200
30 Correct 15 ms 348 KB ans=YES N=190
31 Correct 6 ms 348 KB ans=YES N=187
32 Correct 12 ms 600 KB ans=YES N=187
33 Correct 13 ms 348 KB ans=YES N=182
34 Correct 16 ms 552 KB ans=YES N=184
35 Correct 21 ms 348 KB ans=YES N=188
36 Correct 18 ms 348 KB ans=YES N=181
37 Correct 23 ms 348 KB ans=YES N=188
38 Correct 19 ms 624 KB ans=YES N=191
39 Correct 3 ms 460 KB ans=YES N=196
40 Correct 3 ms 348 KB ans=YES N=196
41 Correct 3 ms 516 KB ans=YES N=196
42 Correct 3 ms 520 KB ans=YES N=196
43 Correct 19 ms 348 KB ans=YES N=195

Subtask #3
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 0 ms 348 KB ans=NO N=4
4 Correct 0 ms 348 KB ans=YES N=5
5 Correct 0 ms 348 KB ans=YES N=9
6 Correct 0 ms 348 KB ans=YES N=5
7 Correct 0 ms 348 KB ans=NO N=9
8 Correct 0 ms 348 KB ans=NO N=10
9 Correct 0 ms 348 KB ans=YES N=10
10 Correct 0 ms 348 KB ans=YES N=10
11 Correct 0 ms 348 KB ans=YES N=10
12 Correct 0 ms 348 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 0 ms 348 KB ans=YES N=8
15 Correct 1 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
17 Correct 0 ms 348 KB ans=YES N=17
18 Correct 0 ms 348 KB ans=YES N=25
19 Correct 2 ms 348 KB ans=YES N=100
20 Correct 9 ms 344 KB ans=YES N=185
21 Correct 1 ms 348 KB ans=NO N=174
22 Correct 1 ms 348 KB ans=YES N=90
23 Correct 1 ms 344 KB ans=YES N=63
24 Correct 1 ms 348 KB ans=YES N=87
25 Correct 5 ms 348 KB ans=YES N=183
26 Correct 7 ms 348 KB ans=YES N=188
27 Correct 10 ms 348 KB ans=YES N=183
28 Correct 6 ms 348 KB ans=YES N=189
29 Correct 9 ms 348 KB ans=YES N=200
30 Correct 15 ms 348 KB ans=YES N=190
31 Correct 6 ms 348 KB ans=YES N=187
32 Correct 12 ms 600 KB ans=YES N=187
33 Correct 13 ms 348 KB ans=YES N=182
34 Correct 16 ms 552 KB ans=YES N=184
35 Correct 21 ms 348 KB ans=YES N=188
36 Correct 18 ms 348 KB ans=YES N=181
37 Correct 23 ms 348 KB ans=YES N=188
38 Correct 19 ms 624 KB ans=YES N=191
39 Correct 3 ms 460 KB ans=YES N=196
40 Correct 3 ms 348 KB ans=YES N=196
41 Correct 3 ms 516 KB ans=YES N=196
42 Correct 3 ms 520 KB ans=YES N=196
43 Correct 19 ms 348 KB ans=YES N=195
44 Correct 8 ms 5724 KB ans=NO N=1934
45 Correct 4 ms 820 KB ans=NO N=1965
46 Execution timed out 3591 ms 944 KB Time limit exceeded
47 Halted 0 ms 0 KB -

Subtask #4
0 / 17.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 5828 KB ans=NO N=1934
2 Correct 5 ms 604 KB ans=NO N=1965
3 Execution timed out 3574 ms 860 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #5
0 / 20.0

# 결과 실행 시간 메모리 Grader output
1 Correct 0 ms 344 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 0 ms 348 KB ans=NO N=4
4 Correct 0 ms 348 KB ans=YES N=5
5 Correct 0 ms 348 KB ans=YES N=9
6 Correct 0 ms 348 KB ans=YES N=5
7 Correct 0 ms 348 KB ans=NO N=9
8 Correct 0 ms 348 KB ans=NO N=10
9 Correct 0 ms 348 KB ans=YES N=10
10 Correct 0 ms 348 KB ans=YES N=10
11 Correct 0 ms 348 KB ans=YES N=10
12 Correct 0 ms 348 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 0 ms 348 KB ans=YES N=8
15 Correct 1 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
17 Correct 0 ms 348 KB ans=YES N=17
18 Correct 0 ms 348 KB ans=YES N=25
19 Correct 2 ms 348 KB ans=YES N=100
20 Correct 9 ms 344 KB ans=YES N=185
21 Correct 1 ms 348 KB ans=NO N=174
22 Correct 1 ms 348 KB ans=YES N=90
23 Correct 1 ms 344 KB ans=YES N=63
24 Correct 1 ms 348 KB ans=YES N=87
25 Correct 5 ms 348 KB ans=YES N=183
26 Correct 7 ms 348 KB ans=YES N=188
27 Correct 10 ms 348 KB ans=YES N=183
28 Correct 6 ms 348 KB ans=YES N=189
29 Correct 9 ms 348 KB ans=YES N=200
30 Correct 15 ms 348 KB ans=YES N=190
31 Correct 6 ms 348 KB ans=YES N=187
32 Correct 12 ms 600 KB ans=YES N=187
33 Correct 13 ms 348 KB ans=YES N=182
34 Correct 16 ms 552 KB ans=YES N=184
35 Correct 21 ms 348 KB ans=YES N=188
36 Correct 18 ms 348 KB ans=YES N=181
37 Correct 23 ms 348 KB ans=YES N=188
38 Correct 19 ms 624 KB ans=YES N=191
39 Correct 3 ms 460 KB ans=YES N=196
40 Correct 3 ms 348 KB ans=YES N=196
41 Correct 3 ms 516 KB ans=YES N=196
42 Correct 3 ms 520 KB ans=YES N=196
43 Correct 19 ms 348 KB ans=YES N=195
44 Correct 8 ms 5724 KB ans=NO N=1934
45 Correct 4 ms 820 KB ans=NO N=1965
46 Execution timed out 3591 ms 944 KB Time limit exceeded
47 Halted 0 ms 0 KB -

Subtask #6
0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2668 ms 15344 KB ans=NO N=66151
2 Correct 2503 ms 10560 KB ans=NO N=64333
3 Execution timed out 3523 ms 20292 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #7
0 / 19.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 5828 KB ans=NO N=1934
2 Correct 5 ms 604 KB ans=NO N=1965
3 Execution timed out 3574 ms 860 KB Time limit exceeded
4 Halted 0 ms 0 KB -