Submission #649789

#	Time	Username	Problem	Language	Result	Execution time	Memory
649789		birthdaycake	Commuter Pass (JOI18_commuter_pass)	C++17	31 / 100	729 ms	30796 KiB

commuter_pass

#ifndef _GLIBCXX_NO_ASSERT
#include <cassert>
#endif
#include <cctype>
#include <cerrno>
#include <cfloat>
#include <ciso646>
#include <climits>
#include <clocale>
#include <cmath>
#include <csetjmp>
#include <csignal>
#include <cstdarg>
#include <cstddef>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
 
#if __cplusplus >= 201103L
#include <ccomplex>
#include <cfenv>
#include <cinttypes>
#include <cstdbool>
#include <cstdint>
#include <ctgmath>
#include <cwchar>
#include <cwctype>
#endif
 
// C++
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <utility>
#include <valarray>
#include <vector>
 
#if __cplusplus >= 201103L
#include <array>
#include <atomic>
#include <chrono>
#include <condition_variable>
#include <forward_list>
#include <future>
#include <initializer_list>
#include <mutex>
#include <random>
#include <ratio>
#include <regex>
#include <scoped_allocator>
#include <system_error>
#include <thread>
#include <tuple>
#include <typeindex>
#include <type_traits>
#include <unordered_map>
#include <unordered_set>
#endif
 
 
 
 
// - - - - - -
 
 
// #include<bits/stdc++.h>
using namespace std;
#define int long long
 
vector<pair<int,int>>adj[200001];
int ans = 1e18;
int dij[5][100001];
int best[100001];
vector<int>a1[100001];
vector<int>a2[100001];
signed main(){
    int n,m; cin >> n >> m;
    int s,t; cin >> s >> t;
    int u,v; cin >> u >> v;
    for(int i = 0; i < m; i++){
        int a,b,c; cin >> a >> b >> c;
        adj[a].push_back({b,c});
        adj[b].push_back({a,c});
    }
    set<pair<int,int>>d;
    for(int i = 1; i <= n; i++){
        dij[0][i] = dij[1][i] = dij[2][i] = dij[3][i] = 1e18;
    }
    dij[0][s] = 0;
    d.insert({dij[0][s], s});
    while(d.size()){
        auto s = *d.begin();
        d.erase(s);
        if(dij[0][s.second] < s.first) continue;
        dij[0][s.second] = s.first;
        for(auto x:adj[s.second]){
            if(dij[0][s.second] + x.second < dij[0][x.first]){
                best[x.first] = s.second;
                dij[0][x.first] = dij[0][s.second] + x.second;
                d.insert({dij[0][x.first], x.first});
            }
        }
    }
    dij[1][t] = 0;
    d.insert({dij[1][t], t});
    while(d.size()){
        auto s = *d.begin();
        d.erase(s);
        if(dij[1][s.second] < s.first) continue;
        dij[1][s.second] = s.first;
        for(auto x:adj[s.second]){
            if(dij[1][s.second] + x.second < dij[1][x.first]){
                dij[1][x.first] = dij[1][s.second] + x.second;
                d.insert({dij[1][x.first], x.first});
            }
        }
    }
    dij[2][v] = 0;
    d.insert({dij[2][v], v});
    while(d.size()){
        auto s = *d.begin();
        d.erase(s);
        if(dij[2][s.second] < s.first) continue;
        dij[2][s.second] = s.first;
        for(auto x:adj[s.second]){
            if(dij[2][s.second] + x.second < dij[2][x.first]){
                dij[2][x.first] = dij[2][s.second] + x.second;
                d.insert({dij[2][x.first], x.first});
            }
        }
    }
    dij[3][u] = 0;
    d.insert({dij[3][u], u});
    while(d.size()){
        auto s = *d.begin();
        d.erase(s);
        if(dij[3][s.second] < s.first) continue;
        dij[3][s.second] = s.first;
        for(auto x:adj[s.second]){
            if(dij[3][s.second] + x.second < dij[3][x.first]){
                dij[3][x.first] = dij[3][s.second] + x.second;
                d.insert({dij[3][x.first], x.first});
            }
        }
    }
    ans = dij[3][v];
    vector<pair<int,int>>a1,a2;
    for(int i = 1; i <= n; i++){
        if(dij[0][i] + dij[1][i] != dij[0][t]) continue;
        a1.push_back({dij[3][i],i});
        a2.push_back({dij[2][i],i});
    }
    sort(a1.begin(), a1.end());
    sort(a2.begin(), a2.end());
    for(int i = 0; i < a2.size(); i++){
        if(i != (a2.size() - 1) && a2[i].second == a1[i].second){
            ans = min(ans,a1[i].first + a2[i + 1].first);
            ans = min(ans,a1[i + 1].first + a2[i].first);

        }else{
            ans = min(ans,a1[i].first + a2[i].first);
        }
    }
    
    
    cout << ans;
}

Compilation message (stderr)

commuter_pass.cpp: In function 'int main()':
commuter_pass.cpp:182:22: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  182 |     for(int i = 0; i < a2.size(); i++){
      |                    ~~^~~~~~~~~~~
commuter_pass.cpp:183:14: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  183 |         if(i != (a2.size() - 1) && a2[i].second == a1[i].second){
      |            ~~^~~~~~~~~~~~~~~~~~

• Subtask 1: 16 / 16
• Subtask 2: 15 / 15
• Subtask 3: 0 / 24
• Subtask 4: 0 / 45