結果

問題 No.1690 Power Grid
ユーザー publfl
提出日時 2021-09-24 22:34:14
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 93 ms / 3,000 ms
コード長 1,792 bytes
コンパイル時間 632 ms
コンパイル使用メモリ 61,168 KB
実行使用メモリ 91,236 KB
最終ジャッジ日時 2024-07-05 10:59:34
合計ジャッジ時間 2,521 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <stdio.h>
#include <vector>
#include <queue>
#define MAX (long long int)1e16

int a;
std::vector< std::pair<int,int> > V[110];
long long int dist[110][110];

std::priority_queue< std::pair<long long int,int> > Q;
void init(int target)
{
	for(int i=1;i<=a;i++) dist[target][i] = MAX;
	Q.push(std::make_pair(0,target));
	while(!Q.empty())
	{
		std::pair<long long int,int> A = Q.top();
		Q.pop();
		A.first *= (-1);
		if(dist[target][A.second]!=MAX) continue;
		dist[target][A.second] = A.first;
		
		for(int i=0;i<V[A.second].size();i++)
		{
			Q.push(std::make_pair(-(A.first+V[A.second][i].second),V[A.second][i].first));
		}
	}
}

int x[110];
long long int dist2[21][300010];
void init2(int k, int bit, long long int min, int target)
{
	if(k==0)
	{
		dist2[target][bit] = min;
		return;
	}
	
	init2(k-1,bit,min,target);
	init2(k-1,bit^(1<<(k-1)),min<dist[target][k]?min:dist[target][k],target);
}

long long int check[21][300010];
long long int func(int k, int bit)
{
	if(k==0) return 0;
	if(check[k][bit]!=MAX) return check[k][bit];
	
	long long int ans = MAX;
	for(int i=1;i<=a;i++)
	{
		if(((bit>>(i-1))&1)==0)
		{
			long long int t = func(k-1,bit+(1<<(i-1))) + dist2[i][bit] + x[i];
			ans = ans<t?ans:t;
		}
	}
	return check[k][bit] = ans;
}

int main()
{
	int b,c;
	scanf("%d%d%d",&a,&b,&c);
	for(int i=1;i<=a;i++) scanf("%d",&x[i]);
	for(int i=1;i<=b;i++)
	{
		int d,e,f;
		scanf("%d%d%d",&d,&e,&f);
		V[d].push_back(std::make_pair(e,f));
		V[e].push_back(std::make_pair(d,f));
	}
	
	for(int i=1;i<=a;i++) init(i);
	for(int i=1;i<=a;i++) init2(a,0,MAX,i);
	
	for(int i=1;i<=c;i++) for(int j=0;j<(1<<a);j++) check[i][j] = MAX;
	
	long long int ans = MAX;
	for(int i=1;i<=a;i++)
	{
		long long int t = func(c-1,1<<(i-1)) + x[i];
		ans = ans<t?ans:t;
	}
	printf("%lld",ans);
}
0