aboutsummaryrefslogtreecommitdiff
path: root/store/works/life/algorithm-experiment/3.1.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'store/works/life/algorithm-experiment/3.1.cpp')
-rw-r--r--store/works/life/algorithm-experiment/3.1.cpp38
1 files changed, 38 insertions, 0 deletions
diff --git a/store/works/life/algorithm-experiment/3.1.cpp b/store/works/life/algorithm-experiment/3.1.cpp
new file mode 100644
index 0000000..1796dc3
--- /dev/null
+++ b/store/works/life/algorithm-experiment/3.1.cpp
@@ -0,0 +1,38 @@
+#include <dlib/optimization/max_cost_assignment.h>
+#include <iostream>
+#include <vector>
+
+#include <dlib/optimization.h>
+
+int main() {
+ int n;
+ std::cin >> n;
+
+ dlib::matrix<int> matrix(n, n);
+
+ for (int i = 0; i < n; i++) {
+ for (int j = 0; j < n; j++) {
+ int l;
+ std::cin >> l;
+ matrix(i, j) = -l;
+ }
+ }
+
+ auto result = dlib::max_cost_assignment(matrix);
+
+ for (auto i : result) {
+ std::cout << i << ' ';
+ }
+
+ std::cout << '\n';
+
+ int sum = 0;
+
+ for (int i = 0; i < n; i++) {
+ sum += matrix(i, result[i]);
+ }
+
+ std::cout << sum;
+
+ return 0;
+}