結果

問題 No.1690 Power Grid
コンテスト
ユーザー kotatsugame
提出日時 2021-09-24 21:40:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 76 ms / 3,000 ms
コード長 918 bytes
コンパイル時間 632 ms
コンパイル使用メモリ 73,844 KB
実行使用メモリ 5,504 KB
最終ジャッジ日時 2024-07-05 10:16:14
合計ジャッジ時間 2,770 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:11:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   11 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
int N,M,K;
int A[18];
long dp[1<<18];
long dist[18][18];
long tmp[18];
main()
{
	cin>>N>>M>>K;
	for(int i=0;i<N;i++)cin>>A[i];
	for(int i=0;i<N;i++)for(int j=0;j<N;j++)if(i!=j)dist[i][j]=1e18;
	for(int i=0;i<M;i++)
	{
		int u,v,w;cin>>u>>v>>w;
		u--,v--;
		dist[u][v]=dist[v][u]=w;
	}
	for(int k=0;k<N;k++)for(int i=0;i<N;i++)for(int j=0;j<N;j++)dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j]);
	for(int i=0;i<1<<N;i++)dp[i]=1e18;
	dp[0]=0;
	long ans=1e18;
	for(int i=0;i<1<<N;i++)
	{
		if(__builtin_popcount(i)==K)
		{
			ans=min(ans,dp[i]);
		}
		for(int j=0;j<N;j++)tmp[j]=1e18;
		for(int j=0;j<N;j++)if(i>>j&1)
		{
			for(int k=0;k<N;k++)tmp[k]=min(tmp[k],dist[j][k]);
		}
		for(int j=0;j<N;j++)if(!(i>>j&1))
		{
			long nxt=dp[i]+A[j];
			if(i)nxt+=tmp[j];
			dp[i|1<<j]=min(dp[i|1<<j],nxt);
		}
	}
	cout<<ans<<endl;
}
0