#define _CRT_SECURE_NO_WARNINGS
#include <bits/stdc++.h>
//#include <ext/pb_ds/tree_policy.hpp>
//#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
//using namespace __gnu_pbds;
#define FOR(i,start,end) for(int i=start;i<(int)(end);i++)
#define FORE(i,start,end) for(int i=start;i<=(int)end;i++)
#define RFOR(i,start,end) for(int i = start; i>end; i--)
#define RFORE(i,start,end) for(int i = start; i>=end; i--)
#define all(a) a.begin(), a.end()
#define mt make_tuple
#define v vector
#define sf scanf
#define pf printf
#define dvar(x) cout << #x << " := " << x << "\n"
#define darr(x,n) FOR(i,0,n) cout << #x << "[" << i << "]" << " := " << x[i] << "\n"
typedef long long ll;
typedef long double ld;
typedef pair<int, int > pii;
typedef pair<ll, ll> pll;
const ll MOD = 1000000007LL;
const ll PRIME = 105943LL;
const int INF = 1e6;
//template <class T> using ordered_set = tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update>;
template<class T> void minn(T &a, T b) { a = min(a, b); }
template<class T> void maxx(T &a, T b) { a = max(a, b); }
void io() {
#ifdef LOCAL_PROJECT
freopen("input.in", "r", stdin); freopen("output.out", "w", stdout);
#else
/* online submission */
#endif
ios_base::sync_with_stdio(false); cin.tie(NULL);
}
/*************************COCI 16-17 R3 #3*************************/
int __builtin_popcount(int s) {
int cnt = 0;
FORE(i, 0, 31) if (s&(1 << i)) cnt++;
return cnt;
}
int N, K;
int cost[20][20];
int getNode(int dist[], bool inSet[]) {
pii best{ INT_MAX, -1 };
FOR(i, 0, N) if (!inSet[i])
minn(best, { dist[i], i });
return best.second;
}
int calc(int bitmask) {
int dist[20];
memset(dist, INF, sizeof dist);
bool inSet[20];
memset(inSet, 0, sizeof inSet);
int par[20];
memset(par, -1, sizeof par);
FOR(i, 0, N) if (bitmask&(1 << i)) dist[i] = 0;
FOR(c, 0, N) {
int u = getNode(dist, inSet);
assert(u >= 0);
inSet[u] = 1;
FOR(i, 0, N) if (!inSet[i] && dist[i] > cost[i][u]) {
dist[i] = cost[i][u];
par[i] = u;
}
}
int ans = 0;
FOR(i, 0, N) if (par[i] != -1)
ans += cost[i][par[i]];
return ans;
}
int main() {
io();
cin >> N >> K;
FOR(i, 0, N) FOR(j, 0, N)
cin >> cost[i][j];
int ans = INT_MAX;
FOR(s, 0, 1 << N) if (__builtin_popcount(s) == K) {
minn(ans, calc(s));
}
cout << ans << "\n";
return 0;
}
/*
...
*/
Compilation message
kronican.cpp: In function 'int __builtin_popcount(int)':
kronican.cpp:38:5: warning: new declaration 'int __builtin_popcount(int)' ambiguates built-in declaration 'int __builtin_popcount(unsigned int)' [-Wbuiltin-declaration-mismatch]
int __builtin_popcount(int s) {
^~~~~~~~~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
376 KB |
Output is correct |
2 |
Correct |
2 ms |
376 KB |
Output is correct |
3 |
Correct |
2 ms |
408 KB |
Output is correct |
4 |
Incorrect |
2 ms |
548 KB |
Output isn't correct |
5 |
Incorrect |
7 ms |
548 KB |
Output isn't correct |
6 |
Incorrect |
7 ms |
632 KB |
Output isn't correct |
7 |
Incorrect |
27 ms |
632 KB |
Output isn't correct |
8 |
Incorrect |
44 ms |
632 KB |
Output isn't correct |
9 |
Incorrect |
45 ms |
632 KB |
Output isn't correct |
10 |
Correct |
331 ms |
632 KB |
Output is correct |