제출 #871293

#제출 시각아이디문제언어결과실행 시간메모리
871293aaron_dcoderCommuter Pass (JOI18_commuter_pass)C++17
100 / 100
435 ms43600 KiB
#define NDEBUG


#ifdef NDEBUG
#define dbg(TXTMSG) if constexpr (false) cerr << "lol"
#define dbgv(VARN) ((void)0)
#define dbgfor(COND) if constexpr (false) for (COND)

#else
#define _GLIBCXX_DEBUG 1
#define _GLIBCXX_DEBUG_PEDANTIC 1
#pragma GCC optimize("trapv")
#define dbg(TXTMSG) cerr << "\n" << TXTMSG
#define dbgv(VARN) cerr << "\n" << #VARN << " = "<< VARN << ", line: " << __LINE__ << "\n"
#define dbgfor(COND) for (COND)

#endif

#include <bits/stdc++.h>
using namespace std;
using ll = long long; 
using pll = pair<ll,ll>;
#define e0 first
#define e1 second
constexpr ll INFTY = 2e18;

ll N,M,S,T,U,V;

			//{to,wi}
vector<vector<pll>> road; 

			//to
vector<vector<ll>> dag_out;
//from
vector<vector<ll>> dag_in;

vector<ll> Vdist,Udist;

struct stat {
	ll cost;
	ll to;
	ll from;

	bool operator <(const stat& oth) const {
		return cost > oth.cost;
	}
};

void djiktra(ll start, vector<ll>* costs) {
	priority_queue<stat> dijkstraq;
	dijkstraq.push({0,start,-1});
	
	while (!dijkstraq.empty()) {
		auto [cost, curr, _] = dijkstraq.top();
		dijkstraq.pop();

		if ((*costs)[curr] != -1) {
			assert((*costs)[curr]<=cost);
			continue;
		}
		(*costs)[curr]=cost;

		for (auto [next, wi] : road[curr])
		{
			dijkstraq.push({cost+wi, next, -1});
		}
	}

}

vector<bool> goodnode;

void dfsgood(ll node) {
	if (goodnode[node]) throw "up";
	goodnode[node]=true;
	for (ll next : dag_in[node])
	{
		if (!goodnode[next]) dfsgood(next);
	}
}

vector<ll> bestVdist;
vector<ll> seen;

ll dfsvcost(ll node) {
	if (!goodnode[node]) throw "up";

	if (seen[node]==2) {
		assert(bestVdist[node]!=0);
		return bestVdist[node];
	}
	else if (seen[node]!=0) throw "up";

	seen[node]=1;
	bestVdist[node]=Vdist[node];
	for (ll next : dag_out[node])
	{
		if (goodnode[next]) {
			bestVdist[node] = min(bestVdist[node], dfsvcost(next));
		}
	}
	seen[node]=2;
	return bestVdist[node];
}


vector<ll> bestUdist;

ll dfsucost(ll node) {
	if (!goodnode[node]) throw "up";

	if (seen[node]==2) {
		assert(bestUdist[node]!=0);
		return bestUdist[node];
	}
	else if (seen[node]!=0) throw "up";

	seen[node]=1;
	bestUdist[node]=Udist[node];
	for (ll next : dag_out[node])
	{
		if (goodnode[next]) {
			bestUdist[node] = min(bestUdist[node], dfsucost(next));
		} 
	}
	seen[node]=2;
	return bestUdist[node];
}

int main() {
	cin.tie(nullptr);
	ios_base::sync_with_stdio(false);

	cin >> N >> M >> S >> T >> U >> V;
	S--; T--; U--; V--; // cb

	road.assign(N,{});
	dag_out.assign(N,{});
	dag_in.assign(N,{});
	Vdist.assign(N,-1);
	Udist.assign(N,-1);
	goodnode.assign(N,false);
	bestVdist.assign(N,-1);
	bestUdist.assign(N,-1);
	seen.assign(N,0);

	for (ll i = 0; i < M; ++i)
	{
		ll A,B,C;
		cin >> A >> B >> C;
		A--; B--; //c
		//#warning
		road[A].push_back({B,C});
		road[B].push_back({A,C});
	}

	{
	vector<ll> node_cost(N,-1);
	dbg("hmm0");
	
	priority_queue<stat> dijkstraq;
	dijkstraq.push({0,S,-1});
	
	while (!dijkstraq.empty()) {
		auto [cost, curr, prev] = dijkstraq.top();
		dijkstraq.pop();
		//dbg(curr <<":" <<cost);
		if (node_cost[curr] != -1 && cost > node_cost[curr]) {
			continue;
		}
		//dbgv(curr <<":" <<cost);


		assert(node_cost[curr]==cost || node_cost[curr]==-1);
		

		if (prev!=-1) {
			dag_out[prev].push_back(curr);
			dag_in[curr].push_back(prev);
		}

		if (node_cost[curr] == -1) 
		{
			node_cost[curr] = cost;
			for (auto [next, wi] : road[curr])
			{
				dijkstraq.push({cost+wi, next, curr});
			}
		}

	}
	dbgv(node_cost[T]);
	}
	//throw exception();
	dbg("hmm1");

	dfsgood(T);
dbg("hmm2");

	djiktra(V,&Vdist);
	djiktra(U,&Udist);

	seen.assign(N,0);
	dfsucost(S);
	seen.assign(N,0);
	dfsvcost(S);

	ll outp=Vdist[U];
	assert(Vdist[U]==Udist[V]);
//	ll o2=INFTY;
dbg("hmm3");
	/*for (ll i = 0; i < N; ++i)
	{
		dbg(i << ":" << Vdist[i] << " " << Udist[i] << " " << goodnode[i]);
		if (goodnode[i]) {
		outp = min(outp,Vdist[i]);	
		o2 = min(o2,Udist[i]);	
		} 
	}*/

	for (ll i = 0; i < N; ++i)
	{
				dbg(i << ":" << Vdist[i] << " " << Udist[i] << " " << goodnode[i]);
				dbg(i << ":" << bestVdist[i] << " " << bestUdist[i] << "=");

				dbgv(bestUdist[i]+Vdist[i]);
		if (goodnode[i]) {outp = min({outp, bestUdist[i]+Vdist[i], bestVdist[i]+Udist[i]});}
	}

	cout << outp;




}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...