Submission #900544

# Submission time Handle Problem Language Result Execution time Memory
900544 2024-01-08T13:44:21 Z gun_gan Dreaming (IOI13_dreaming) C++17
18 / 100
36 ms 15056 KB
#include "dreaming.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int MX = 1e5 + 5;
int N, M, L;
vector<pair<int,int>> g[MX];

bool vis[MX];
int cur = 0, mx = 0, z = 0, len = 0;
vector<int> vec;
int p = 0, q = 0;
pair<int,int> par[MX];

void dfs(int v) {
      vis[v] = 1;
      vec.push_back(v);
      if(cur > mx) {
            mx = cur;
            z = v;
      }
      for(auto [u, w] : g[v]) {
            if(!vis[u]) {
                  if(cur + w > len / 2 && !p && len > 0) {
                        p = cur;
                        q = len - cur;
                  }
                  cur += w;
                  par[u] = {v, w};
                  dfs(u);
                  cur -= w;
            }
      }
}

vector<pair<int,int>> P;

int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
      for(int i = 0; i < M; i++) {
            g[A[i]].push_back({B[i], T[i]});
            g[B[i]].push_back({A[i], T[i]});
      }

      for(int i = 1; i <= N; i++) {
            if(vis[i]) continue;
            vec.clear();
            cur = 0, mx = 0, z = 0, len = 0;

            dfs(i);
            for(auto x : vec) {
                  vis[x] = 0;
            }

            if(vec.size() == 1) {
                  for(auto x : vec) {
                        vis[x] = 1;
                  }
                  P.push_back({0, 0});
                  continue;
            }

            if(vec.size() == 2) {
                  for(auto x : vec) {
                        vis[x] = 1;
                  }
                  P.push_back({mx, 0});
                  continue;
            }

            vec.clear();
            int tz = z;
            cur = 0, mx = 0, z = 0;
            dfs(tz);

            len = mx;

            int k = z, sum = 0;
            while(k != tz) {
                  if(sum + par[k].second > len / 2 && sum <= len / 2) {
                        p = sum;
                        q = len - sum;
                  }
                  sum += par[k].second;
                  k = par[k].first;
            }

            vec.clear();

            P.push_back({max(p, q), min(p, q)});
      }

      sort(P.rbegin(), P.rend());

      int ans = 0;

      for(auto [x, y] : P) {
            ans = max(ans, x + y);
      }

      for(int i = 1; i < P.size(); i++) {
            auto [x, y] = P[i];
            ans = max(ans, P[0].first + L + x);
      }

      if(P.size() > 2) {
            ans = max(ans, P[1].first + 2 * L + P[2].first);
      }

      return ans;
}

Compilation message

dreaming.cpp: In function 'int travelTime(int, int, int, int*, int*, int*)':
dreaming.cpp:101:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  101 |       for(int i = 1; i < P.size(); i++) {
      |                      ~~^~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 30 ms 15056 KB Output is correct
2 Correct 36 ms 14956 KB Output is correct
3 Correct 21 ms 11736 KB Output is correct
4 Correct 5 ms 6236 KB Output is correct
5 Correct 4 ms 5464 KB Output is correct
6 Correct 8 ms 7004 KB Output is correct
7 Incorrect 1 ms 4700 KB Output isn't correct
8 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 4444 KB Output is correct
2 Correct 1 ms 4444 KB Output is correct
3 Incorrect 1 ms 4444 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 30 ms 15056 KB Output is correct
2 Correct 36 ms 14956 KB Output is correct
3 Correct 21 ms 11736 KB Output is correct
4 Correct 5 ms 6236 KB Output is correct
5 Correct 4 ms 5464 KB Output is correct
6 Correct 8 ms 7004 KB Output is correct
7 Incorrect 1 ms 4700 KB Output isn't correct
8 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 17 ms 7648 KB Output is correct
2 Correct 14 ms 7644 KB Output is correct
3 Correct 16 ms 7636 KB Output is correct
4 Correct 14 ms 7644 KB Output is correct
5 Correct 15 ms 7600 KB Output is correct
6 Correct 15 ms 8412 KB Output is correct
7 Correct 14 ms 7904 KB Output is correct
8 Correct 14 ms 7608 KB Output is correct
9 Correct 14 ms 7620 KB Output is correct
10 Correct 14 ms 7892 KB Output is correct
11 Correct 1 ms 4696 KB Output is correct
12 Correct 4 ms 5844 KB Output is correct
13 Correct 4 ms 5844 KB Output is correct
14 Correct 4 ms 5844 KB Output is correct
15 Correct 4 ms 5844 KB Output is correct
16 Correct 4 ms 5844 KB Output is correct
17 Correct 4 ms 5844 KB Output is correct
18 Correct 5 ms 5676 KB Output is correct
19 Correct 4 ms 5844 KB Output is correct
20 Correct 1 ms 4444 KB Output is correct
21 Correct 1 ms 4444 KB Output is correct
22 Correct 1 ms 4952 KB Output is correct
23 Correct 4 ms 5844 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 4444 KB Output is correct
2 Correct 1 ms 4444 KB Output is correct
3 Incorrect 1 ms 4444 KB Output isn't correct
4 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 30 ms 15056 KB Output is correct
2 Correct 36 ms 14956 KB Output is correct
3 Correct 21 ms 11736 KB Output is correct
4 Correct 5 ms 6236 KB Output is correct
5 Correct 4 ms 5464 KB Output is correct
6 Correct 8 ms 7004 KB Output is correct
7 Incorrect 1 ms 4700 KB Output isn't correct
8 Halted 0 ms 0 KB -