結果

問題 No.1690 Power Grid
ユーザー snrnsidy
提出日時 2021-10-25 10:52:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 132 ms / 3,000 ms
コード長 1,401 bytes
コンパイル時間 2,857 ms
コンパイル使用メモリ 193,548 KB
最終ジャッジ日時 2025-01-25 06:47:25
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

long long int n,m,k,a,b,c;
long long int dp[1<<18];
long long int dist[18][18];
long long int A[18];

int main(void)
{
	cin.tie(0);
	ios::sync_with_stdio(false);

	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] = 0;
			}
			else
			{
				dist[i][j] = 1e18;
			}
		}
	}


	for(int i=0;i<m;i++)
	{
		cin >> a >> b >> c;
		a-=1;
		b-=1;
		dist[a][b] = min(dist[a][b],c);
		dist[b][a] = min(dist[b][a],c);
	}

	for(int k=0;k<n;k++)
	{
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			{
				if(dist[i][j] > dist[i][k] + dist[k][j])
				{
					dist[i][j] = dist[i][k] + dist[k][j];
				}
			}
		}
	}

	for(int i=0;i<(1<<n);i++)
	{
		dp[i] = 1e18;
	}

	dp[0] = 0;

	for(int i=0;i<(1<<n);i++)
	{
		for(int j=0;j<n;j++)
		{
			if((i&(1<<j))) continue;
			long long int val = 1e18;
			for(int k=0;k<n;k++)
			{
				if((i&(1<<k)))
				{
					val = min(val,dist[j][k]);
				}
			}
			if(val>=1e18) val = 0;
			//cout << j << ' ' << val << ' ' << A[j] << '\n';
			dp[i | (1<<j)] = min(dp[i | (1<<j)],dp[i] + val + A[j]);
		}
	}

	long long int res = 1e18;

	for(int i=0;i<(1<<n);i++)
	{
		int cnt = 0;
		for(int j=0;j<n;j++)
		{
			if((i&(1<<j))) cnt++;
		}
		if(cnt==k)
		{
			res = min(res,dp[i]);
		}
	}

	cout << res << '\n';
	return 0;
}
0