Submission #1095963

# Submission time Handle Problem Language Result Execution time Memory
1095963 2024-10-03T13:24:04 Z yoav_s Building Skyscrapers (CEOI19_skyscrapers) C++17
22 / 100
3500 ms 22684 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;

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;
    }
    for (ll i = n - 1; i >= 0; i--)
    {
        bool done = false;
        for (ll x : available)
        {
            vv supplyComp(maxX, v(maxY));
            vvvp newSupplyGraph(maxX, vvp(maxY));
            occupied[cells[x].f][cells[x].s] = false;
            for (ll i = 0; i < maxX; i++)
            {
                for (ll j = 0; j < maxY; j++)
                {
                    if (occupied[i][j]) continue;
                    if (i + 1 < maxX && !occupied[i + 1][j])
                    {
                        newSupplyGraph[i][j].eb(i+1,j);
                        newSupplyGraph[i+1][j].eb(i,j);
                    }
                    if (j + 1 < maxY && !occupied[i][j + 1])
                    {
                        newSupplyGraph[i][j].eb(i,j+1);
                        newSupplyGraph[i][j+1].eb(i,j);
                    }
                }
            }
            vv component(maxX, v(maxY, -1));
            ll comp = 0;
            for (ll i = 0; i < maxX; i++)
            {
                for (ll j = 0; j < maxY; j++)
                {
                    if (component[i][j] != -1) continue;
                    stack<p> dfs;
                    dfs.emplace(i, j);
                    while (!dfs.empty())
                    {
                        auto top = dfs.top();
                        dfs.pop();
                        if (component[top.f][top.s] != -1) continue;
                        component[top.f][top.s] = comp;
                        for (auto x : newSupplyGraph[top.f][top.s]) dfs.push(x);
                    }
                    comp++;
                }
            }
            bool poss = true;
            if (component[cells[x].f][cells[x].s] != component[maxX - 1][maxY - 1]) 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;
            }
            available.erase(x);
            order.pb(x);
            done = true;
            break;
        }
        if (!done)
        {
            cout << "NO\n";
            assert(false);
            return 0;
        }
    }
    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:51: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]
   51 |     for (ll i = 0; i < relevantX.size(); i++) xShrink[relevantX[i]] = i + 1;
      |                    ~~^~~~~~~~~~~~~~~~~~
skyscrapers.cpp:52: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]
   52 |     for (ll i = 0; i < relevantY.size(); i++) yShrink[relevantY[i]] = i + 1;
      |                    ~~^~~~~~~~~~~~~~~~~~
skyscrapers.cpp:152: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]
  152 |                 if (cnt != available.size() - 1)
      |                     ~~~~^~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 1 ms 344 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 1 ms 348 KB ans=YES N=10
10 Correct 1 ms 348 KB ans=YES N=10
11 Correct 0 ms 456 KB ans=YES N=10
12 Correct 1 ms 344 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 1 ms 348 KB ans=YES N=8
15 Correct 0 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 1 ms 344 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 1 ms 348 KB ans=YES N=10
10 Correct 1 ms 348 KB ans=YES N=10
11 Correct 0 ms 456 KB ans=YES N=10
12 Correct 1 ms 344 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 1 ms 348 KB ans=YES N=8
15 Correct 0 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
17 Correct 1 ms 348 KB ans=YES N=17
18 Correct 1 ms 348 KB ans=YES N=25
19 Correct 15 ms 348 KB ans=YES N=100
20 Correct 86 ms 552 KB ans=YES N=185
21 Correct 1 ms 344 KB ans=NO N=174
22 Correct 14 ms 524 KB ans=YES N=90
23 Correct 40 ms 344 KB ans=YES N=63
24 Correct 15 ms 344 KB ans=YES N=87
25 Correct 62 ms 348 KB ans=YES N=183
26 Correct 128 ms 576 KB ans=YES N=188
27 Correct 193 ms 596 KB ans=YES N=183
28 Correct 95 ms 596 KB ans=YES N=189
29 Correct 186 ms 596 KB ans=YES N=200
30 Correct 353 ms 616 KB ans=YES N=190
31 Correct 108 ms 596 KB ans=YES N=187
32 Correct 312 ms 604 KB ans=YES N=187
33 Correct 545 ms 848 KB ans=YES N=182
34 Correct 827 ms 604 KB ans=YES N=184
35 Correct 2509 ms 984 KB ans=YES N=188
36 Correct 834 ms 692 KB ans=YES N=181
37 Correct 1318 ms 740 KB ans=YES N=188
38 Correct 3258 ms 1116 KB ans=YES N=191
39 Correct 62 ms 344 KB ans=YES N=196
40 Correct 63 ms 348 KB ans=YES N=196
41 Correct 62 ms 344 KB ans=YES N=196
42 Correct 64 ms 568 KB ans=YES N=196
43 Correct 1586 ms 852 KB ans=YES N=195
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 1 ms 344 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 1 ms 348 KB ans=YES N=10
10 Correct 1 ms 348 KB ans=YES N=10
11 Correct 0 ms 456 KB ans=YES N=10
12 Correct 1 ms 344 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 1 ms 348 KB ans=YES N=8
15 Correct 0 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
17 Correct 1 ms 348 KB ans=YES N=17
18 Correct 1 ms 348 KB ans=YES N=25
19 Correct 15 ms 348 KB ans=YES N=100
20 Correct 86 ms 552 KB ans=YES N=185
21 Correct 1 ms 344 KB ans=NO N=174
22 Correct 14 ms 524 KB ans=YES N=90
23 Correct 40 ms 344 KB ans=YES N=63
24 Correct 15 ms 344 KB ans=YES N=87
25 Correct 62 ms 348 KB ans=YES N=183
26 Correct 128 ms 576 KB ans=YES N=188
27 Correct 193 ms 596 KB ans=YES N=183
28 Correct 95 ms 596 KB ans=YES N=189
29 Correct 186 ms 596 KB ans=YES N=200
30 Correct 353 ms 616 KB ans=YES N=190
31 Correct 108 ms 596 KB ans=YES N=187
32 Correct 312 ms 604 KB ans=YES N=187
33 Correct 545 ms 848 KB ans=YES N=182
34 Correct 827 ms 604 KB ans=YES N=184
35 Correct 2509 ms 984 KB ans=YES N=188
36 Correct 834 ms 692 KB ans=YES N=181
37 Correct 1318 ms 740 KB ans=YES N=188
38 Correct 3258 ms 1116 KB ans=YES N=191
39 Correct 62 ms 344 KB ans=YES N=196
40 Correct 63 ms 348 KB ans=YES N=196
41 Correct 62 ms 344 KB ans=YES N=196
42 Correct 64 ms 568 KB ans=YES N=196
43 Correct 1586 ms 852 KB ans=YES N=195
44 Correct 8 ms 5724 KB ans=NO N=1934
45 Correct 4 ms 604 KB ans=NO N=1965
46 Execution timed out 3568 ms 1072 KB Time limit exceeded
47 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 9 ms 5720 KB ans=NO N=1934
2 Correct 5 ms 604 KB ans=NO N=1965
3 Execution timed out 3573 ms 1116 KB Time limit exceeded
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB ans=YES N=1
2 Correct 0 ms 348 KB ans=YES N=4
3 Correct 1 ms 344 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 1 ms 348 KB ans=YES N=10
10 Correct 1 ms 348 KB ans=YES N=10
11 Correct 0 ms 456 KB ans=YES N=10
12 Correct 1 ms 344 KB ans=YES N=9
13 Correct 0 ms 348 KB ans=YES N=9
14 Correct 1 ms 348 KB ans=YES N=8
15 Correct 0 ms 348 KB ans=YES N=8
16 Correct 0 ms 348 KB ans=NO N=2
17 Correct 1 ms 348 KB ans=YES N=17
18 Correct 1 ms 348 KB ans=YES N=25
19 Correct 15 ms 348 KB ans=YES N=100
20 Correct 86 ms 552 KB ans=YES N=185
21 Correct 1 ms 344 KB ans=NO N=174
22 Correct 14 ms 524 KB ans=YES N=90
23 Correct 40 ms 344 KB ans=YES N=63
24 Correct 15 ms 344 KB ans=YES N=87
25 Correct 62 ms 348 KB ans=YES N=183
26 Correct 128 ms 576 KB ans=YES N=188
27 Correct 193 ms 596 KB ans=YES N=183
28 Correct 95 ms 596 KB ans=YES N=189
29 Correct 186 ms 596 KB ans=YES N=200
30 Correct 353 ms 616 KB ans=YES N=190
31 Correct 108 ms 596 KB ans=YES N=187
32 Correct 312 ms 604 KB ans=YES N=187
33 Correct 545 ms 848 KB ans=YES N=182
34 Correct 827 ms 604 KB ans=YES N=184
35 Correct 2509 ms 984 KB ans=YES N=188
36 Correct 834 ms 692 KB ans=YES N=181
37 Correct 1318 ms 740 KB ans=YES N=188
38 Correct 3258 ms 1116 KB ans=YES N=191
39 Correct 62 ms 344 KB ans=YES N=196
40 Correct 63 ms 348 KB ans=YES N=196
41 Correct 62 ms 344 KB ans=YES N=196
42 Correct 64 ms 568 KB ans=YES N=196
43 Correct 1586 ms 852 KB ans=YES N=195
44 Correct 8 ms 5724 KB ans=NO N=1934
45 Correct 4 ms 604 KB ans=NO N=1965
46 Execution timed out 3568 ms 1072 KB Time limit exceeded
47 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 3141 ms 15304 KB ans=NO N=66151
2 Correct 2702 ms 10680 KB ans=NO N=64333
3 Execution timed out 3571 ms 22684 KB Time limit exceeded
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 9 ms 5720 KB ans=NO N=1934
2 Correct 5 ms 604 KB ans=NO N=1965
3 Execution timed out 3573 ms 1116 KB Time limit exceeded
4 Halted 0 ms 0 KB -