제출 #1252349

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1252349		yf_yusuf	통행료 (APIO13_toll)	C++20	0 / 100	150 ms	51124 KiB

toll

//```//    YF YUSUF
#define Free_open(File) freopen((File".in"),"r",stdin);freopen((File".out"),"w",stdout);
#define YF ios_base::sync_with_stdio(0);cout.setf(ios::fixed);
#define YUSUF cout.precision(0);cout.tie(0);cin.tie(0);
#include <bits/stdc++.h>
#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include <unordered_map>
#include <cmath>
#include <numeric>
#include <queue>
#include <stack>
#include <cassert>
#include <climits>
#include <string>
using namespace std;
#ifdef YF_CHECK
	bool Output=1;
//	#include "g/debug.h"
//  #include "g/bigint.h"
//	#include "g/text.h"
#else
	#pragma GCC optimize ("unroll-loops")
	#pragma GCC optimize ("inline")
	#pragma GCC optimize ("Ofast")
	#pragma GCC optimize ("O3")
	bool Output=0;
	#define deb(x...) 42 
#endif
#ifdef DEV
	bool LOUT=1;
#else
	bool LOUT=0;
#endif
using ll   = long long;
using ld   = long double;
using sll  = set <ll>;
using vll  = vector <ll>;
using mll  = map <ll,ll>;
using pll  = pair <ll,ll>;
using vvl  = vector <vll>;
using vpll = vector <pll>;
template<class T>T MIN(T&a,T b){a=min(a,b);return a;}
template<class T>T MAX(T&a,T b){a=max(a,b);return a;}
#define revers(a) reverse(all(a))
#define all(a) a.begin(),a.end()
#define sortt(a) sort(all(a))
#define sgr v+v+1,(tl+tr)/2+1,tr
#define sgl v+v,tl,(tl+tr)/2
#define pb push_back
#define ins insert
#define S second
#define F first
ll BP(ll a,ll b,ll mod=1e9+7){
	if(b==0)return 1;
	ll q=BP(a,b/2,mod);
	return ((q*q)%mod*(b%2?a:1ll))%mod;
}
ll f(ll x){return x*(x+1)/2;}
ll dup(ll a,ll b){return (a+b-1)/b;}
ll lcm(ll a,ll b){return a/__gcd(a,b)*b;}
ll invf(ll x){return (-1+sqrt(1+8*x))/2;}
const ll  mod=998244353;
const ll  INF=1e18;
const ll  inf=1e9+7;
const ll  N  =1e6+7;
vll a(N);
vpll KK,g[N];
ll ans,n,m,k;
struct DSU{
	vll p,sz;
	vvl g; 
	vll b, d, pr;
	void build(ll n){
		p.resize(n+1, 0);
		sz.resize(n+1, 1);
		g.resize(n+1,(vll){});
		b.resize(n+1, 0);
		d.resize(n+1, 0);
		pr.resize(n+1,0);
		for(int i=1;i<=n;i++){
			p[i] = i;
			sz[i] = 1;
		}
	}
	ll get(ll x){
		if(p[x]==x)return x;
		else return p[x] = get(p[x]);
	}
	bool check(ll a,ll b){
		a = get(a);
		b = get(b);
		return (a!=b);
	}
	bool up(ll a,ll b){
		ll A = a, B = b;
		a = get(a);
		b = get(b);
		if(a==b)return 0;
		if(sz[a] < sz[b])swap(a,b);
		p[b] = a;
		sz[a] += sz[b];
		sz[b] = 0;
		g[A].pb(B);
		g[B].pb(A);
		return 1;
	}
	void dp(ll v=1,ll p=1){
		for(auto to : g[v]){
			if(to!=p){
				d[to] = d[v] + 1;
				pr[to] = v;
				dp(to, v);
			}
		}
	}
	void dfs(ll v,ll p){
		b[v] = a[v];
		for(auto to : g[v])if(to!=p){
			dfs(to, v);
			b[v] += b[to];
		}
	}
};
ll used[N];
map<pll,ll> W;
ll mx;
void dist(ll v,ll fn,ll ret){
	if(v == fn)
		MIN(mx, ret);
	used[v] = 1;
	for(auto [w, to] : g[v])if(!used[to]){
		dist(to, fn, max(ret, w));
	}
	used[v] = 0;
}
void YF_MAIN(ll TEST){
	cin>>n>>m>>k;
	set<vll> e;
	ll wk[k]{};
	for(int i=1;i<=m;i++){
		ll v,u,w;
		cin>>v>>u>>w;
		W[{v,u}] = W[{u, v}] = w;
		g[v].pb({w,u});
		g[u].pb({w,v});
		e.ins({w,1,u,v});
	}
	DSU MST;
	MST.build(n);
	for(auto V : e)
		MST.up(V[2], V[3]);
	MST.dp();
	for(int i=1;i<=k;i++){
		ll v,u;
		cin>>v>>u;
		KK.pb({v,u});
		mx = inf;
		dist(v, u, 0);
		wk[i-1] = mx;
	}
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	ll cost = 0;
	for(int mask = 1; mask < (1<<k); mask++){
		vpll K;
		auto E = e;
		map<pll,bool>mp;
		for(int i=0;i<k;i++){
			if((mask >> i) & 1){
				K.pb(KK[i]);
				// cout<<wk[i]<<" WK\n";
				E.ins({wk[i], 0, KK[i].F, KK[i].S});
				mp[{KK[i].F, KK[i].S}] = 1;
			}
		}
		DSU dsu;
		dsu.build(n);
		bool can = 1;
		for(auto V : E){
			ll v = V[2], u = V[3];
			bool B = dsu.up(v, u);
			if(mp[{v,u}]){
				can &= B;
				// cout<<v<<" "<<u<<" M\n";
			}
			// cout<<B<<" "<<v<<" "<<u<<" M\n";
		}
		if(!can)continue;
		dsu.dfs(1, 1);
		ll sum = 0;
		for(int i=0;i<k;i++){
			if((mask >> i) & 1){
				auto [v,u] = KK[i];
				if(v == dsu.pr[u])swap(v,u);
				sum += dsu.b[u] * wk[i];
			}
		}
		// cout<<mask<<"\n";
		MAX(cost, sum);
	}
	cout<<cost;
}
bool TECT=0;
bool CASE=0;
bool OUTP=1;
const ll LN=1e1 + 7; ll lg[LN];
const ll SN=2e1 + 7; ll SM[SN];
const ll FN=2e6 + 7; 
ll FACT[FN],inv[FN],FMOD=inf;
ll PER(ll n,ll k){return FACT[n] *inv[n-k]%FMOD;}
ll CNK(ll n,ll k){return PER(n,k)*inv[k  ]%FMOD;}
void BEFORE(){
	for(ll i=2;i<SN;i++){
		if(SM[i])continue;
		for(ll j=i;j<SN;j+=i)
			MAX(SM[j],i);
	}
	for(int i=2;i<LN;i++) lg[i]=lg[i/2]+1;
	FACT[0]=inv[0]=1;
	for(int i=1;i<FN;i++){
		FACT[i]=FACT[i-1]*i%FMOD;
		if(i<5e5+7)inv[i]=BP(FACT[i],FMOD-2,FMOD);
	}
}
signed main(){
	if(OUTP){YF YUSUF}
	// freopen(("input.txt"),"r",stdin);freopen(("output.txt"),"w",stdout);
	// Free_open("TEST");
	cout<<(LOUT && Output ? "\nYF_OUTPUT:\n\n" : "");
	int TEST=1;
	if(TECT)
		cin>>TEST;
	BEFORE();
	for(int T=1;T<=TEST;T++){
		if(CASE)
			cout<<"Case "<<T<<": ";
		YF_MAIN(T);
		cout<<(T==TEST ? "" : "\n");
	}
	return 0;
} 
//```

• Subtask 1: 0 / 16
• Subtask 2: 0 / 18
• Subtask 3: 0 / 22
• Subtask 4: 0 / 22
• Subtask 5: 0 / 22