Submission #535423

# Submission time Handle Problem Language Result Execution time Memory
535423 2022-03-10T09:16:55 Z AngusWong Dreaming (IOI13_dreaming) C++17
18 / 100
43 ms 16280 KB
#include "dreaming.h"
#include <bits/stdc++.h>
#define pii pair<int, int>
#define f first
#define s second
using namespace std;

int n, m, l, dis[100001], leaf[100001], len, stp, edp, ans;
vector<pii> v[100001];
vector<int> path, mx;

void dfs(int x, int prev=-1){
    dis[x]=0;
    leaf[x]=x;
    int mx=x, mx2=x;
    for (pii i: v[x]){
        if (i.f==prev) continue;
        dfs(i.f, x);
        dis[i.f]+=i.s;
        if (dis[i.f]>dis[mx]) mx2=mx, mx=i.f;
        else if (dis[i.f]>dis[mx2]) mx2=i.f;
    }
    if (dis[mx]+dis[mx2]>len) len=dis[mx]+dis[mx2], stp=leaf[mx], edp=leaf[mx2];
    dis[x]=dis[mx];
    leaf[x]=leaf[mx];
}

bool get_path(int x, int prev=-1){
    if (x==edp) return true;
    for (pii i: v[x]){
        if (i.f==prev) continue;
        if (get_path(i.f, x)){
            path.push_back(i.s);
            return true;
        }
    }
    return false;
}

int max_length(int x){
    len=stp=edp=-1;
    dfs(x);
    path.clear();
    get_path(stp);
    if (path.empty()) return 0;
    for (int i=1; i<path.size(); i++) path[i]+=path[i-1];
    int ans=2e9;
    for (int i=0; i<path.size(); i++){
        int t=path[i]; // 0..i
        if (i<path.size()) t=max(t, path[path.size()-1]-path[i]);
        ans=min(ans, t);
    }
    return ans;
}

int travelTime(int N, int M, int L, int A[], int B[], int T[]) {
    n=N, m=M, l=L;
    for (int i=0; i<m; i++){
        v[A[i]].push_back({B[i], T[i]});
        v[B[i]].push_back({A[i], T[i]});
    }
    for (int i=0; i<n; i++) dis[i]=-1;
    for (int i=0; i<n; i++){
        if (dis[i]==-1) mx.push_back(max_length(i));
    }
    sort(mx.begin(), mx.end(), greater<int>());
    ans=mx[0];
    if (mx.size()>1) ans=max(ans, mx[0]+mx[1]+l);
    if (mx.size()>2) ans=max(ans, mx[1]+mx[2]+l*2);
    return ans;
}

Compilation message

dreaming.cpp: In function 'int max_length(int)':
dreaming.cpp:46:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |     for (int i=1; i<path.size(); i++) path[i]+=path[i-1];
      |                   ~^~~~~~~~~~~~
dreaming.cpp:48:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   48 |     for (int i=0; i<path.size(); i++){
      |                   ~^~~~~~~~~~~~
dreaming.cpp:50:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   50 |         if (i<path.size()) t=max(t, path[path.size()-1]-path[i]);
      |             ~^~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Incorrect 43 ms 16280 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 43 ms 16280 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 18 ms 6488 KB Output is correct
2 Correct 24 ms 6460 KB Output is correct
3 Correct 21 ms 6488 KB Output is correct
4 Correct 24 ms 6500 KB Output is correct
5 Correct 19 ms 6512 KB Output is correct
6 Correct 20 ms 7000 KB Output is correct
7 Correct 23 ms 6636 KB Output is correct
8 Correct 21 ms 6468 KB Output is correct
9 Correct 17 ms 6356 KB Output is correct
10 Correct 21 ms 6612 KB Output is correct
11 Correct 1 ms 2644 KB Output is correct
12 Correct 5 ms 4072 KB Output is correct
13 Correct 5 ms 4048 KB Output is correct
14 Correct 5 ms 4048 KB Output is correct
15 Correct 6 ms 4116 KB Output is correct
16 Correct 6 ms 4048 KB Output is correct
17 Correct 5 ms 3944 KB Output is correct
18 Correct 6 ms 4176 KB Output is correct
19 Correct 5 ms 4048 KB Output is correct
20 Correct 2 ms 2668 KB Output is correct
21 Correct 1 ms 2664 KB Output is correct
22 Correct 2 ms 2616 KB Output is correct
23 Correct 4 ms 4048 KB Output is correct
# Verdict Execution time Memory Grader output
1 Incorrect 1 ms 2644 KB Output isn't correct
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Incorrect 43 ms 16280 KB Output isn't correct
2 Halted 0 ms 0 KB -