CSES - NOI 2024 - Results
Submission details
Task:Anime Shops
Sender:Teo Lovmar
Submission time:2024-03-06 18:14:47 +0200
Language:C++ (C++11)
Status:READY
Result:0
Feedback
groupverdictscore
#10
#20
#30
Test results
testverdicttimegroup
#10.00 s1, 3details
#20.00 s1, 3details
#30.00 s1, 3details
#40.01 s1, 3details
#50.11 s3details
#60.01 s3details
#70.01 s3details
#80.01 s3details
#90.05 s2, 3details
#100.01 s2, 3details
#110.01 s2, 3details
#120.01 s3details
#130.01 s3details
#140.01 s3details
#150.01 s3details
#160.05 s3details

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
#define fi first
#define se second

ll n, m, k;
bool marked[10000] = {};
set<ll> marks;
vector<vector<ll>> adj;

void group1() {
  for (ll start = 0; start < n; start++) {
    deque<ll> tovis;
    set<ll> vis;
    ll dist = -1;
    ll depth = 0;
    tovis.push_back(start);
    tovis.push_back(-1);
    vis.insert(start);
    while (1 < tovis.size() && dist == -1) {
      ll node = *tovis.begin();
      tovis.pop_front();
      if (node == -1) {
        tovis.push_back(-1);
        depth++;
        continue;
      }

      if (node != start && marked[node])
        dist = depth;

      for (auto nbr : adj[node]) {
        if (vis.find(nbr) != vis.end())
          continue;
        tovis.push_back(nbr);
        vis.insert(node);
      }
    }

    cout << dist << "\n";
  }

  cout << endl;
}

void group2() {
  for (ll i = 0; i < n; i++) {
    if (marked[i])
      marks.erase(i);

    if (marks.size() == 0) {
      cout << "-1" << endl;
      if (marked[i])
        marks.insert(i);
      continue;
    }

    ll min_dist = LLONG_MAX;
    auto ub = marks.upper_bound(i);
    if (ub != marks.end()) {
      min_dist = abs((*ub) - i);
    }

    if (ub != marks.begin()) {
      auto lb = prev(ub);
      min_dist = min(abs(i - (*lb)), min_dist);
    }

    cout << min_dist << "\n";

    if (marked[i])
      marks.insert(i);
  }

  cout << endl;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(NULL);

  cin >> n >> m >> k;

  for (ll i = 0; i < k; i++) {
    ll mark;
    cin >> mark;
    marked[mark - 1] = 1;
    marks.insert(mark - 1);
  }

  adj.resize(n);
  for (ll i = 0; i < m; i++) {
    ll a, b;
    cin >> a >> b;
    a--;
    b--;
    adj[a].push_back(b);
    adj[b].push_back(a);
  }

  // if ((string(getenv("AS_GRP")).compare("1")))
  //  group1();
  // else
  group2();

  return 0;
}

Test details

Test 1

Group: 1, 3

Verdict:

input
1000 2000 1
816
1 868
976 995
377 536
...

correct output
4 3 3 4 6 4 -1 4 5 2 2 5 5 3 5...

user output
815
814
813
812
811
...
Truncated

Test 2

Group: 1, 3

Verdict:

input
1000 2000 20
578 955 222 813 494 962 753 71...

correct output
5 6 4 3 4 2 -1 3 3 3 4 3 2 3 -...

user output
204
203
202
201
200
...
Truncated

Test 3

Group: 1, 3

Verdict:

input
1000 2000 100
945 230 119 680 975 520 490 28...

correct output
2 2 3 -1 2 -1 3 2 2 1 2 -1 3 2...

user output
8
7
6
5
4
...
Truncated

Test 4

Group: 1, 3

Verdict:

input
1000 2000 1000
150 443 960 545 218 487 896 38...

correct output
1 -1 1 1 -1 1 1 1 -1 1 1 1 -1 ...

user output
1
1
1
1
1
...
Truncated

Test 5

Group: 3

Verdict:

input
100000 200000 1
77222
59470 61811
43092 48939
14395 19964
...

correct output
8 10 8 8 8 8 8 8 9 7 7 8 8 8 6...

user output
77221
77220
77219
77218
77217
...
Truncated

Test 6

Group: 3

Verdict:

input
100000 200000 20
70440 82302 64483 96767 51485 ...

correct output
-1 8 6 5 5 7 6 7 6 6 8 5 6 4 5...

user output
(empty)

Test 7

Group: 3

Verdict:

input
100000 200000 100
68738 37820 55519 77758 46482 ...

correct output
4 5 5 5 5 4 6 -1 5 4 5 6 4 5 5...

user output
(empty)

Test 8

Group: 3

Verdict:

input
100000 200000 100000
47137 80137 73347 78145 9205 6...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
(empty)

Test 9

Group: 2, 3

Verdict:

input
100000 99999 1
44158
1 2
2 3
3 4
...

correct output
44157 44156 44155 44154 44153 ...

user output
44157
44156
44155
44154
44153
...
Truncated

Test 10

Group: 2, 3

Verdict:

input
100000 99999 20
44158 25720 84658 90057 99607 ...

correct output
361 360 359 358 357 356 355 35...

user output
(empty)

Test 11

Group: 2, 3

Verdict:

input
100000 99999 100000
44158 25720 84658 90057 99607 ...

correct output
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...

user output
(empty)

Test 12

Group: 3

Verdict:

input
100000 99999 3
99998 99999 100000
1 2
1 3
1 4
...

correct output
33333 33332 33332 33332 33331 ...

user output
(empty)

Test 13

Group: 3

Verdict:

input
100000 99999 300
99701 99702 99703 99704 99705 ...

correct output
333 333 333 333 333 333 333 33...

user output
(empty)

Test 14

Group: 3

Verdict:

input
100000 99999 30000
70001 70002 70003 70004 70005 ...

correct output
3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 ...

user output
(empty)

Test 15

Group: 3

Verdict:

input
100000 0 100000
1 2 3 4 5 6 7 8 9 10 11 12 13 ...

correct output
-1 -1 -1 -1 -1 -1 -1 -1 -1 -1 ...

user output
(empty)

Test 16

Group: 3

Verdict:

input
100000 100000 2
1 50001
1 2
2 3
3 4
...

correct output
50000 1 2 3 4 5 6 7 8 9 10 11 ...

user output
50000
1
2
3
4
...
Truncated