Submission #380271

#	Time	Username	Problem	Language	Result	Execution time	Memory
380271		ritul_kr_singh	Kronican (COCI16_kronican)	C++17	10 / 100	219 ms	364 KiB

kronican

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define sp << ' ' <<
#define nl << '\n'
 
vector<int> e;
int get(int u){
	return e[u] < 0 ? u : e[u] = get(e[u]);
}
bool unite(int u, int v){
	u = get(u), v = get(v);
	if(u==v) return false;
	if(e[u] > e[v]) swap(u, v);
	e[u] += e[v], e[v] = u;
	return true;
}
 
signed main(){
	ios_base::sync_with_stdio(false); cin.tie(nullptr);
	int n, k; cin >> n >> k;
	int d[n][n];
	vector<array<int, 3>> edges;
	for(int i=0; i<n; ++i){
		for(int j=0; j<n; ++j){
			cin >> d[i][j];
		}
	}
	for(int i=0; i<n; ++i){
		for(int j=i+1; j<n; ++j){
			edges.push_back({min(d[i][j], d[j][i]), i, j});
		}
	}
	sort(edges.begin(), edges.end());
 
	int ans = 1e18;
	for(int i=0; i<(1<<n); ++i){
		int cnt = 0;
		for(int j=0; j<n; ++j) cnt += (bool)((1<<j)&i);
		if(cnt != k) continue;
		e.assign(n, -1);
		int first = -1, res = 0;
		for(int j=0; j<n; ++j) if((1<<j)&i) first = j;
		for(int j=0; j<n; ++j) if((1<<j)&i) unite(j, first);
		for(auto edge : edges){
			if(unite(edge[1], edge[2])) res += edge[0];
		}
		ans = min(ans, res);
	}
	cout << ans;
}

• Subtask 1: 10 / 100