Submission #527792

#	Time	Username	Problem	Language	Result	Execution time	Memory
527792		1ne	Autobus (COCI22_autobus)	C++14	30 / 70	1097 ms	64628 KiB

Main

#include<bits/stdc++.h>
using namespace std;
 
int main(){
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n,m;cin>>n>>m;
vector<vector<int64_t>>dist(n,vector<int64_t>(n,1e15));
vector<vector<set<pair<int64_t,int64_t>>>>arr(n,vector<set<pair<int64_t,int64_t>>>(n));
for (int i = 0;i<m;++i){
	int64_t x,y,z;cin>>x>>y>>z;
	--x;
	--y;
	dist[x][y] = min(dist[x][y],z);
}
for (int i = 0;i<n;++i){
	for (int j = 0;j<n;++j){
		if (i!=j&&dist[i][j]!=1e15){
			arr[i][j].insert({dist[i][j],1});
		}
	}
}
 
 
int kk,q;cin>>kk>>q;
kk = min(kk,n*n);
for (int i = 0;i<n;++i){
	for (int j = 0;j<n;++j){
		for (int k = 0;k<n;++k){
			int64_t mindist = 1e15,mink = 1e15;
			vector<pair<int64_t,int64_t>>temp;
			for (auto x:arr[j][i]){
				if (x.first<mindist){
					mindist = x.first;
				}
				else if (x.second<mink){
					mink = x.second;
				}
				else temp.push_back(x);
			}
			for (auto x:temp){
				arr[j][i].erase(x);
			}
			temp.clear();
			mindist=1e15;
			mink=1e15;
			for (auto x:arr[i][k]){
				if (x.first<mindist){
					mindist = x.first;
				}
				else if (x.second<mink){
					mink = x.second;
				}
				else temp.push_back(x);
			}
			for (auto x:temp){
				arr[i][k].erase(x);
			}
			for (auto x:arr[j][i]){
				for (auto y:arr[i][k]){
					if (x.second + y.second<=kk)
					arr[j][k].insert({x.first + y.first,x.second + y.second});
				}
			}
		}
	}
}
for (int i = 0;i<q;++i){
	int x,y;cin>>x>>y;
	--x;
	--y;
	if (x==y){
		cout<<0<<'\n';
		continue;
	}
	if (arr[x][y].empty()){
		cout<<-1<<'\n';
	}
	else{
		auto z = *arr[x][y].begin();
		cout<<z.first<<'\n';
	}
}
return 0;}

• Subtask 1: 15 / 15
• Subtask 2: 15 / 15
• Subtask 3: 0 / 25
• Subtask 4: 0 / 15