結果

問題 No.2604 Initial Motion
ユーザー TairitsuMeowTairitsuMeow
提出日時 2024-01-12 22:05:49
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,664 bytes
コンパイル時間 2,479 ms
コンパイル使用メモリ 194,796 KB
実行使用メモリ 340,824 KB
最終ジャッジ日時 2024-01-12 22:06:04
合計ジャッジ時間 7,632 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,676 KB
testcase_01 AC 2 ms
6,676 KB
testcase_02 AC 2 ms
6,676 KB
testcase_03 AC 61 ms
12,236 KB
testcase_04 AC 60 ms
11,600 KB
testcase_05 AC 56 ms
11,924 KB
testcase_06 AC 60 ms
12,180 KB
testcase_07 AC 60 ms
12,392 KB
testcase_08 AC 61 ms
12,208 KB
testcase_09 AC 57 ms
11,788 KB
testcase_10 AC 60 ms
12,016 KB
testcase_11 AC 62 ms
12,268 KB
testcase_12 AC 62 ms
12,208 KB
testcase_13 TLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

// Problem: No.2604 Initial Motion
// Contest: yukicoder
// URL: https://yukicoder.me/problems/no/2604
// Memory Limit: 512 MB
// Time Limit: 3000 ms

#include<bits/stdc++.h>
#include <atcoder/mincostflow>
#define debug(x) cerr<<(#x)<<" "<<(x)<<endl
using namespace atcoder;
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define pii pair<ll,ll>
#define rep(i,a,b) for(ll i=(a);i<=(b);++i)
#define per(i,a,b) for(ll i=(a);i>=(b);--i)
using namespace std;
ll read(){
	ll x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void write(ll x){
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
}
const ll N=2e3+5,INF=1e18;
ll n,m,S,T;
ll dis[N],ok[N],a[N],b[N],k;
vector<pii>to[N];
int main(){
	k=read(),n=read(),m=read();
	rep(i,1,k)a[i]=read();
	rep(i,1,n)b[i]=read();
	rep(i,1,m){
		ll x=read(),y=read(),z=read();
		to[x].push_back(make_pair(y,z));
		to[y].push_back(make_pair(x,z));
	}
    mcf_graph<ll, ll> g(n+k+2);
    ll S = 0, T= n+k+1;
	rep(i,1,k){
		rep(j,1,n)dis[j]=INF,ok[j]=0;
		dis[a[i]]=0;
		priority_queue<pii>q;
		q.push(make_pair(0,a[i]));
		while(!q.empty()){
			pii now=q.top();q.pop();
			ll u=now.second;
			if(ok[u])continue;
			ok[u]=1;
			for(pii es:to[u]){
				ll v=es.first,w=es.second;
				if(dis[v]>dis[u]+w){
					dis[v]=dis[u]+w;
					q.push(make_pair(-dis[v],v));
				}
			}
		}
		rep(j,1,n){
			if(b[j])g.add_edge(i,j+k,1,dis[j]);
		}
		g.add_edge(S,i,1,0);
	}
	
	rep(i,1,n){
		if(b[i])g.add_edge(i+k,T,b[i],0);
	}
    pii result = g.flow(S,T, INF);
	write(result.second);
	return 0;
}
0