oj.uz

Submission #1007756

# Submission time Handle Problem Language Result Execution time Memory
1007756 2024-06-25T12:38:21 Z NintsiChkhaidze Cities (BOI16_cities) C++17
100 / 100
 1473 ms 126192 KB 
#include <bits/stdc++.h>
#define ll long long
#define pb push_back
#define s second
#define f first
#define pii pair <int,int>
#define left h*2,l,(l + r)/2
#define right h*2+1,(l + r)/2 + 1,r
using namespace std;
 
const int N = 2e5 + 5;
const ll inf = 1e18;
int arr[N];
ll dist[7][N],dd[34][N],ddd[N];
ll DD[34][N],val[34][N];
vector <pair <int,ll>> v[N];
 
signed main (){
    ios_base::sync_with_stdio(0),cin.tie(NULL),cout.tie(NULL);
 
    int n,k,m;
    cin>>n>>k>>m;
 
    for (int i = 1; i <= k; i++)
        cin >> arr[i];
 
    for (int i =1; i <= m; i++){
        int a,b; ll c;
        cin>>a>>b>>c;
        v[a].pb({b,c});
        v[b].pb({a,c});
    }
 
	for (int i = 1; i <= k; i++){
		for (int j = 1; j <= n; j++)
			dist[i][j] = inf;
 
		priority_queue<pair <ll,int>> pq;
		dist[i][arr[i]] = 0;
		pq.push({0,arr[i]});
		while (pq.size()){
			ll D = -pq.top().f; int x = pq.top().s;
			pq.pop();
			if (dist[i][x] != D) continue;
 
			for (auto [to,w]: v[x]){
				if (dist[i][to] > D + w){
					dist[i][to] = D + w;
					pq.push({-dist[i][to],to});
				}
			}
		}
	}
    
	ll ans=inf;
	for (int i = 1; i <= n; i++){
		ll cur=0;
		for (int j = 1; j <= k; j++)
			cur += dist[j][i];
		ans=min(ans,cur);
	}
 
	if (k < 4){
		cout<<ans<<endl;
		exit(0);
	}
	
	for (int mask1=0;mask1<(1<<k); mask1++)
		for(int i=1;i<=n;i++)
			DD[mask1][i] = inf;
	
	for (int mask2=0;mask2<(1<<k); mask2++)
		for (int i = 1; i <= n; i++)
			for (int j=0;j<k;j++)
				if (((mask2>>j) & 1)) val[mask2][i] += (dist[j + 1][i]);
 
	for (int mask1 = 0; mask1 < (1<<k); mask1++){
		int cnt=0;
		for (int i=0;i<k;i++)
			cnt += ((mask1 >> i) & 1);
		if (cnt != 2) continue;

		priority_queue <pair <ll,int>> pq;
		for (int i = 1; i <= n; i++){
			dd[mask1][i] = val[mask1][i];
			pq.push({-dd[mask1][i],i});
		}
 
		while (pq.size()){
			ll D = -pq.top().f; int x = pq.top().s;
			pq.pop();
			if (dd[mask1][x] != D) continue;
 
			for (auto [to,w]: v[x]){
				if (dd[mask1][to] > D + w){
					dd[mask1][to] = D + w;
					pq.push({-dd[mask1][to],to});
				}
			}
		}
		
		for (int mask2 = 0; mask2 < (1<<k); mask2++){
			int y = (mask1 & mask2);
			if (y > 0) continue;
 
			for (int i = 1; i <= n; i++){
				ll kk = dd[mask1][i] + val[mask2][i];
				DD[(mask1 | mask2)][i] = min(DD[(mask1 | mask2)][i],kk);
			}
		}
		
	}
 
	for (int mask1=0;mask1<(1<<k); mask1++){
 
		priority_queue <pair <ll,int>> pq;
		for (int i = 1; i <= n; i++)
			ddd[i] = DD[mask1][i],pq.push({-ddd[i],i});
 
		while (pq.size()){
			ll D = -pq.top().f; int x = pq.top().s;
			pq.pop();
			if (ddd[x] != D) continue;
 
			for (auto [to,w]: v[x]){
				if (ddd[to] > D + w){
					ddd[to] = D + w;
					pq.push({-ddd[to],to});
				}
			}
		}
 
		for (int i = 1; i <= n; i++){
			ddd[i] += (val[(((1<<k) - 1) ^ mask1)][i]);
			ans=min(ans,ddd[i]);
		}
	}
 
	cout<<ans;
}

Subtask #1
22.0 / 22.0

# Verdict Execution time Memory Grader output
1 Correct 2 ms 10840 KB Output is correct
2 Correct 1 ms 10840 KB Output is correct
3 Correct 2 ms 12892 KB Output is correct
4 Correct 7 ms 74440 KB Output is correct
5 Correct 7 ms 80732 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 120 ms 24972 KB Output is correct
2 Correct 125 ms 25524 KB Output is correct
3 Correct 44 ms 18000 KB Output is correct
4 Correct 40 ms 21952 KB Output is correct
5 Correct 104 ms 22896 KB Output is correct
6 Correct 38 ms 19824 KB Output is correct
7 Correct 2 ms 12892 KB Output is correct
8 Correct 2 ms 10844 KB Output is correct

Subtask #3
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 74588 KB Output is correct
2 Correct 11 ms 74588 KB Output is correct
3 Correct 10 ms 74584 KB Output is correct
4 Correct 9 ms 74648 KB Output is correct

Subtask #4
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 755 ms 96940 KB Output is correct
2 Correct 738 ms 95080 KB Output is correct
3 Correct 474 ms 87472 KB Output is correct
4 Correct 439 ms 90412 KB Output is correct
5 Correct 126 ms 84160 KB Output is correct
6 Correct 57 ms 85456 KB Output is correct

Subtask #5
26.0 / 26.0

# Verdict Execution time Memory Grader output
1 Correct 1473 ms 126192 KB Output is correct
2 Correct 1456 ms 125232 KB Output is correct
3 Correct 1398 ms 124632 KB Output is correct
4 Correct 1202 ms 117300 KB Output is correct
5 Correct 912 ms 97888 KB Output is correct
6 Correct 242 ms 85412 KB Output is correct
7 Correct 105 ms 84292 KB Output is correct