結果

問題 No.1320 Two Type Min Cost Cycle
ユーザー kotatsugamekotatsugame
提出日時 2020-12-17 03:40:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 626 ms / 2,000 ms
コード長 1,029 bytes
コンパイル時間 936 ms
コンパイル使用メモリ 84,736 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-20 06:08:05
合計ジャッジ時間 10,121 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 57
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:43:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   43 | main()
      | ^~~~

ソースコード

diff #

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
int T,N,M;
int U[2000],V[2000],W[2000];
bool ex[2000];
vector<long>dijkstra(int st,int ct)
{
	vector<vector<pair<int,int> > >G(N);
	for(int i=0;i<M;i++)if(ex[i])
	{
		G[U[i]].push_back(make_pair(V[i],W[i]));
		if(T==0)
		{
			G[V[i]].push_back(make_pair(U[i],W[i]));
		}
	}
	vector<long>ret(N,(long)1e18);
	priority_queue<pair<long,int> >P;
	ret[st]=ct;
	P.push(make_pair(-ct,st));
	while(!P.empty())
	{
		long cost=-P.top().first;
		int u=P.top().second;
		P.pop();
		if(ret[u]<cost)continue;
		for(pair<int,int>e:G[u])
		{
			int v=e.first;
			long nxt=cost+e.second;
			if(ret[v]>nxt)
			{
				ret[v]=nxt;
				P.push(make_pair(-nxt,v));
			}
		}
	}
	return ret;
}
main()
{
	cin>>T>>N>>M;
	for(int i=0;i<M;i++)
	{
		cin>>U[i]>>V[i]>>W[i];
		U[i]--,V[i]--;
	}
	long ans=1e18;
	for(int i=0;i<M;i++)
	{
		for(int j=0;j<M;j++)ex[j]=true;
		ex[i]=false;
		ans=min(ans,dijkstra(V[i],W[i])[U[i]]);
	}
	cout<<(ans<(long)1e18?ans:-1)<<endl;
}
0