Artifact Content
Not logged in

Artifact 8cc09a515372166ff7a31ff44fd3c4bf4a5273fc


#include <iostream>
#include <sstream>
#include <iomanip>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <algorithm>
#include <numeric>
#include <iterator>
#include <functional>
#include <complex>
#include <queue>
#include <stack>
#include <cmath>
#include <cassert>
#include <tuple>
using namespace std;
typedef long long LL;
typedef complex<double> CMP;

typedef int Vert;
typedef LL Cost;
typedef pair<Cost,Vert> edge;
typedef vector<edge> edges;
typedef vector<edges> graph;
const LL INF = 0x3ffffffffffff;

class MaliciousPath { public:
	long long minPath(int N, int K, vector <int> from, vector <int> to, vector <int> cost)
	{
		graph G(N), Gr(N);
		for(int i=0; i<from.size(); ++i) {
			G[from[i]].emplace_back(cost[i], to[i]);
			Gr[to[i]].emplace_back(cost[i], from[i]);
		}

		vector<Cost> d(N, INF); {
			priority_queue<edge, edges, greater<edge>> Q;
			Q.emplace(0, N-1);
			while(!Q.empty()) {
				Cost c = Q.top().first;
				Vert v = Q.top().second;
				Q.pop();
				if(d[v] <= c)
					continue;
				d[v] = c;
				for(auto e: Gr[v]) {
					Cost cc = c + e.first;
					Vert vv = e.second;
					if(d[vv] == INF)
						Q.emplace(cc, vv);
				}
			}
			if(d[0] == INF)
				return -1;
		}

		while(K --> 0) {
			vector<Cost> d2(N, 0);
			for(Vert v=0; v<N-1; ++v) {
				LL worst = 0;
				for(auto e: G[v])
					worst = max(worst, e.first+d[e.second]);
				d2[v] = worst;
			}

			d.assign(N, INF);

			priority_queue<edge, edges, greater<edge>> Q;
			Q.emplace(0, N-1);
			while(!Q.empty()) {
				Cost c = Q.top().first;
				Vert v = Q.top().second;
				Q.pop();
				if(d[v] <= c)
					continue;
				if(c < d2[v])
					continue;
				d[v] = c;
				for(auto e: Gr[v]) {
					Vert vv = e.second;
					Cost cc = max(d2[vv], c + e.first);
					if(d[vv] == INF)
						Q.emplace(cc, vv);
				}
			}
		}

		return d[0];
	}
};

// BEGIN CUT HERE
#include <ctime>
double start_time; string timer()
 { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); }
template<typename T> ostream& operator<<(ostream& os, const vector<T>& v)
 { os << "{ ";
   for(typename vector<T>::const_iterator it=v.begin(); it!=v.end(); ++it)
   os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; }
void verify_case(const long long& Expected, const long long& Received) {
 bool ok = (Expected == Received);
 if(ok) cerr << "PASSED" << timer() << endl;  else { cerr << "FAILED" << timer() << endl;
 cerr << "\to: \"" << Expected << '\"' << endl << "\tx: \"" << Received << '\"' << endl; } }
#define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock();
#define END	 verify_case(_, MaliciousPath().minPath(N, K, from, to, cost));}
int main(){

CASE(0)
	int N = 3; 
	int K = 1000; 
	int from_[] = {0,1,1,2};
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = {1,0,2,2};
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = {3,2,1,1};
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = 5004LL; 
END
CASE(1)
	int N = 4; 
	int K = 1; 
	int from_[] = {0,0,1,1,1,2,2,3};
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = {1,3,0,2,3,2,1,3};
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = {0,100,103,0,0,34,102,33};
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = 100LL; 
END
CASE(2)
	int N = 10; 
	int K = 5; 
	int from_[] = {0,0,1,1,2,2,3,3,4,4,4,5,5,6,6,7,7,8,8,9,9};
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = {1,1,2,2,3,3,4,4,5,5,4,6,6,7,7,8,8,9,9,9,9};
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = {2,10,10,1,2,10,10,1,2,10,100,10,2,1,10,10,2,1,10,10,1};
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = 514LL; 
END
CASE(3)
	int N = 50; 
	int K = 200; 
	int from_[] = {0,13,8,17,3,8,4,21,11,20,2,18,21,2,4,9,17,0,14,10,15,18,1,22,10,14,
19,24,5,5,12,7,7,16,19,13,20,15,23,6,23,9,3,6,16,11,22,24,12,1,25,25,
26,26,27,27,28,28,29,29,30,30,31,31,32,32,33,33,34,34,35,35,36,36,37,
37,38,38,39,39,40,40,41,41,42,42,43,43,44,44,45,45,46,46,47,47,48,48,
49,49,37,9,14,0,33,20,46,26,12,11,2,7,34,19,37,5,2,17,41,16,34,13,18,
35,6,14,16,25,9,10,5,10,7,36,45,3,6,22,32,28,45,40,16,36,28,16,34,1,
9,19,18,6,15,29,12,5,44,33,49,14,40,1,30,21,37,49,1,44,42,6,38,1,31,
40,37,34,35,6,43,29,41,48,17,4,38,26,4,46,43,6,27,30,0,16,40,33,0,42,
41,10,33,47,11,37,49,25,36,20,47,12,28,17,11,17,26,26,37,34,27,17,8,
2,13,43,36,28,1,23,29,40,18,22,0,7,30,23,3,39,5,23,28,38,44,19,43,15,
16,43,5,27,24,25,7,16,38,33,33,1,9,25,47,0,31,30,29,4,36,49,26,6,39,
40,28,39,48,26,2,15,41,42,32,0,35,34,28,30,40,3,33,16,15,41,45,12,33,
35,16,47,34,23};
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = {41,42,17,0,2,7,28,32,31,33,6,42,11,13,7,40,47,21,4,6,19,15,4,18,30,25,
2,10,30,34,1,47,35,23,3,0,9,25,42,21,4,6,47,32,5,40,5,0,8,49,16,29,8,
11,42,33,35,26,27,43,35,6,14,13,44,25,13,42,2,26,17,3,40,31,18,12,24,
37,0,37,15,44,35,40,10,1,35,47,36,33,2,39,23,28,32,0,6,21,33,41,0,19,
16,29,35,16,44,6,18,17,2,46,41,11,27,5,44,1,48,15,43,8,41,33,16,11,45,
47,19,41,14,41,8,24,13,3,44,41,42,30,31,44,21,14,43,48,0,6,25,38,36,14,
36,22,43,15,20,19,37,25,17,44,17,46,8,25,33,19,42,40,42,24,15,31,34,8,
41,25,20,29,2,5,43,28,33,40,31,27,6,21,9,35,8,8,26,13,11,31,4,4,30,34,
35,1,15,11,10,24,15,24,23,16,24,32,9,1,6,17,48,6,35,19,12,5,21,23,25,9,
17,47,19,23,22,35,19,5,1,10,9,41,11,45,0,23,0,11,39,0,17,2,18,4,17,24,0,
10,19,4,36,22,42,18,13,48,27,11,19,28,31,39,32,48,2,26,38,43,38,49,34,
37,11,9,11,14,12,9,37,0,22,14,15,0,8,23,7,43,5,8,16,47};
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = {494,848305,3326,6008,223,2,6,673,152335,25,713909,42842,122,9941,361853,
35,20157,1098,41,83693,365851,43843,622,591230,7722,2489,7,861622,21272,
169,1153,3,7,3,577031,24522,5,241,757900,11036,8892,199,7,2,1,453031,115,
13,125,67,35,91,33427,14,106,203749,5,1804,4543,23,1,281,441212,18,3,
11629,233,388188,10,701,76170,763,875,11606,44972,6449,37409,83516,5912,
6,705899,26759,253,580531,14215,21916,884775,30,678,5940,17,480830,8,
61,218613,683352,4,557589,2619,3658,221515,15825,163577,25,9,1,4,13,2,88,
147,110163,18118,2,15429,211872,24,1,188382,12500,2348,190,4279,40,8428,
56325,933152,231523,9454,21,4855,96168,1722,329515,77,1,3,3518,10926,
12172,4,71,181976,1318,9,5086,905,108490,80164,2,10236,197,1880,17420,
614650,372457,13918,36,17,167,6,254127,512,15,341436,1,186,96,7,3,42,4,
3,4,492598,18523,172302,1,421535,390382,2952,6228,871,505372,131266,5,
743902,11,34,657,4717,3196,259,192504,229,6786,28,44364,21123,8,166781,
885021,2,828,497,376,24707,52,1,1659,19402,27261,82,473,443,1089,586,20,
7,239277,27132,4681,761,10644,17798,1,820306,13330,19,293167,2,4,89279,
14,1,9,18023,1165,495221,32304,538,178613,1,4764,32767,114,103,1,302,428,
92,927352,22270,2646,3599,6,16362,3,4,280286,338,652,2,347022,23,323084,
3338,46,58,1263,93,46992,19112,1,19499,33,807600,20296,16803,911294,3151,
1,793,1,1,504,62209,1397,52726,3650,54,3630,358480,178,394,28,137436,52764,
1209,1599};
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = 121213509LL; 
END
CASE(4)
	int N = 20; 
	int K = 1000; 
	int from_[] = {0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,18};
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = {1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,0,0};
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = {1000000,1000000,1000000,1000000,1000000,1000000,1000000,1000000,1000000,1000000,
1000000,1000000,1000000,1000000,1000000,1000000,1000000,1000000,1000000,1000000,
1000000};
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = 19019000000LL; 
END
CASE(5)
	int N = 2; 
	int K = 0; 
	int from_[] = {0,1};
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = {0,1};
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = {5,4};
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = -1LL; 
END
/*
CASE(6)
	int N = ; 
	int K = ; 
	int from_[] = ;
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = ;
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = ;
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = LL; 
END
CASE(7)
	int N = ; 
	int K = ; 
	int from_[] = ;
	  vector <int> from(from_, from_+sizeof(from_)/sizeof(*from_)); 
	int to_[] = ;
	  vector <int> to(to_, to_+sizeof(to_)/sizeof(*to_)); 
	int cost_[] = ;
	  vector <int> cost(cost_, cost_+sizeof(cost_)/sizeof(*cost_)); 
	long long _ = LL; 
END
*/
}
// END CUT HERE