| # |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
| 544065 |
2022-04-01T00:53:43 Z |
Olympia |
Domino (COCI15_domino) |
C++17 |
|
4000 ms |
524288 KB |
#include <vector>
#include <algorithm>
#include <iostream>
#include <set>
#include <cmath>
#include <map>
#include <random>
#include <cassert>
#include <ctime>
#include <bitset>
#include <stack>
#include <cstdlib>
#include <queue>
#include <stdint.h>
#include <cstdio>
#include <limits.h>
using namespace std;
struct Domino {
pair<int,int> from, to;
int64_t cost;
bool operator < (const Domino& d1) const {
if (this->cost != d1.cost) return (this->cost > d1.cost);
if (this->from != d1.from) return (this->from < d1.from);
if (this->to != d1.to) return (this->to < d1.to);
return false;
}
};
class Graph {
public:
vector<int64_t> weight;
vector<vector<int>> adj;
vector<int64_t> dp;
void add_edge (int u, int v) {
adj[u][v] = adj[v][u] = 1;
}
Graph (int n) {
adj.resize(n);
for (int i = 0; i < n; i++) {
adj[i].assign(n, 0);
}
}
void solve (int k, bool b) {
dp.clear();
dp.assign((1 << (int)adj.size()), 0);
for (int i = 0; i < dp.size(); i++) {
if (__builtin_popcount(i) > k) {
continue;
}
vector<int> nodes;
for (int j = 0; j < adj.size(); j++) {
if (i & (1 << j)) {
nodes.push_back(j);
//cout << "YES ";
}
}
//cout << adj.size() << " " << i << " " << nodes.size() << " " << __builtin_popcount(i) << '\n';
assert(nodes.size() == __builtin_popcount(i));
bool fine = true;
for (int x: nodes) {
for (int y: nodes) {
if (x == y) continue;
if (!adj[x][y]) {
fine = false;
}
}
}
if (!fine) {
continue;
}
for (int x: nodes) {
dp[i] += weight[x];
}
}
if (b) return;
for (int i = 1; i < dp.size(); i++) {
for (int j = 0; j < adj.size(); j++) {
if (i & (1 << j)) {
dp[i] = max(dp[i], dp[i - (1 << j)]);
}
}
}
}
};
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n, m;
cin >> n >> m;
int64_t grid[n][n];
int sm = 0;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
cin >> grid[i][j];
sm += grid[i][j];
}
}
vector<Domino> dominoes;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
for (int dx = -1; dx <= 1; dx++) {
for (int dy = -1; dy <= 1; dy++) {
if (i + dx < 0 || j + dy < 0) continue;
if (i + dx == n || j + dy == n) continue;
if (abs(dx) + abs(dy) != 1) continue;
if (dx < dy) continue;
dominoes.push_back({{i, j}, {i + dx,j + dy}, grid[i][j] + grid[i + dx][j + dy]});
}
}
}
}
sort(dominoes.begin(), dominoes.end());
while (dominoes.size() > 7 * (m - 1) + 5) {
dominoes.pop_back();
}
vector<pair<int64_t,pair<Domino, Domino>>> edges;
for (auto& d1: dominoes) {
for (auto& d2: dominoes) {
set<pair<int,int>> s; s.insert(d1.from), s.insert(d1.to), s.insert(d2.from), s.insert(d2.to);
if (s.size() != 4) continue;
//cout << d1.from.first << " " << d1.from.seco
edges.push_back({d1.cost + d2.cost, {d1, d2}});
}
}
set<Domino> mySet;
for (auto& p: edges) {
mySet.insert(p.second.first), mySet.insert(p.second.second);
}
map<Domino,int> myMap; int cntr = 0;
vector<int64_t> weights;
for (Domino d: mySet) {
weights.push_back(d.cost);
myMap[d] = cntr++;
}
Graph gr(cntr), gr1(cntr/2), gr2(cntr - cntr/2);
gr.weight = weights;
for (int i = 0; i < cntr/2; i++) gr1.weight.push_back(weights[i]);
for (int i = cntr/2; i < cntr; i++) gr2.weight.push_back(weights[i]);
for (auto& e: edges) {
gr.add_edge(myMap[e.second.first], myMap[e.second.second]);
if (myMap[e.second.first] < cntr/2 && myMap[e.second.second] < cntr/2) gr1.add_edge(myMap[e.second.second], myMap[e.second.first]);
if (myMap[e.second.first] >= cntr/2 && myMap[e.second.second] >= cntr/2) gr2.add_edge(myMap[e.second.second] - cntr/2, myMap[e.second.first] - cntr/2);
}
gr1.solve(m, true);
int64_t myMax = 0;
for (int i = 1; i < gr1.dp.size(); i++) {
if (!gr1.dp[i]) continue;
vector<int> nodes;
for (int j = 0; j < gr1.adj.size(); j++) {
if (i & (1 << j)) {
nodes.push_back(j);
}
}
int tot = 0;
for (int j = 0; j < gr2.adj.size(); j++) {
bool fine = true;
for (int k: nodes) {
if (!gr.adj[j + cntr/2][k]) {
fine = false;
break;
}
}
if (fine) {
tot += (1 << j);
}
}
gr2.solve(m - __builtin_popcount(i), false);
//cout << gr2.dp[tot] << " " << gr1.dp[i] << '\n';
myMax = max(gr2.dp[tot] + gr1.dp[i], myMax);
}
gr2.solve(m, false), gr1.solve(m, false);
myMax = max(myMax, max(gr1.dp.back(), gr2.dp.back()));
cout << sm - myMax;
}
Compilation message
domino.cpp: In member function 'void Graph::solve(int, bool)':
domino.cpp:48:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
48 | for (int i = 0; i < dp.size(); i++) {
| ~~^~~~~~~~~~~
domino.cpp:53:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
53 | for (int j = 0; j < adj.size(); j++) {
| ~~^~~~~~~~~~~~
domino.cpp:78:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
78 | for (int i = 1; i < dp.size(); i++) {
| ~~^~~~~~~~~~~
domino.cpp:79:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
79 | for (int j = 0; j < adj.size(); j++) {
| ~~^~~~~~~~~~~~
domino.cpp: In function 'int main()':
domino.cpp:116:28: warning: comparison of integer expressions of different signedness: 'std::vector<Domino>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
116 | while (dominoes.size() > 7 * (m - 1) + 5) {
| ~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~
domino.cpp:150:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
150 | for (int i = 1; i < gr1.dp.size(); i++) {
| ~~^~~~~~~~~~~~~~~
domino.cpp:153:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
153 | for (int j = 0; j < gr1.adj.size(); j++) {
| ~~^~~~~~~~~~~~~~~~
domino.cpp:159:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::vector<int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
159 | for (int j = 0; j < gr2.adj.size(); j++) {
| ~~^~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
87 ms |
14804 KB |
Output is correct |
| 2 |
Correct |
52 ms |
14680 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
2 ms |
596 KB |
Output is correct |
| 2 |
Correct |
1 ms |
596 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1575 ms |
228628 KB |
Output is correct |
| 2 |
Correct |
882 ms |
228808 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
2 ms |
340 KB |
Output is correct |
| 2 |
Correct |
3 ms |
340 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
897 ms |
214992 KB |
Output is correct |
| 2 |
Correct |
525 ms |
215072 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
747 ms |
57492 KB |
Output is correct |
| 2 |
Correct |
338 ms |
61460 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
4103 ms |
228812 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
4097 ms |
17036 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
4103 ms |
236092 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
4054 ms |
198196 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
64 ms |
340 KB |
Output is correct |
| 2 |
Correct |
53 ms |
412 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
4109 ms |
416724 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
200 ms |
524288 KB |
Execution killed with signal 9 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
554 ms |
524288 KB |
Execution killed with signal 9 |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Execution timed out |
4098 ms |
16908 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Runtime error |
1638 ms |
524288 KB |
Execution killed with signal 9 |
| 2 |
Halted |
0 ms |
0 KB |
- |
