#include <iostream>
#include <cmath>
#include <cctype>
#include <vector>
#include <algorithm>
#include <set>
#include <map>
#include <deque>
#include <stack>
#include <unordered_set>
#include <sstream>
#include <cstring>
#include <iomanip>
#include <queue>
#include <unordered_map>
#include <random>
#include <cfloat>
#include <chrono>
#include <bitset>
#include <complex>
#include <immintrin.h>
#include <cassert>
bool try_kuhn(int32_t v, bool* vis, std::vector<int32_t>* graph, int32_t* matched, int32_t left, int32_t right) {
vis[v] = true;
for(int32_t i = 0; i < graph[v].size(); i++)
if(graph[v][i] >= left && graph[v][i] <= right && matched[graph[v][i]] == -1) {
matched[graph[v][i]] = v;
return true;
}
for(int32_t i = 0; i < graph[v].size(); i++)
if(graph[v][i] >= left && graph[v][i] <= right && try_kuhn(matched[graph[v][i]], vis, graph, matched, left, right)) {
matched[graph[v][i]] = v;
return true;
}
return false;
}
int main() {
std::ios::sync_with_stdio(false);
std::cin.tie(0);
int32_t n, m, k;
std::cin >> n >> m >> k;
std::vector<int32_t>* graph = new std::vector<int32_t>[m];
for(int32_t i = 0; i < k; i++) {
int32_t src, dst;
std::cin >> src >> dst;
src--;
dst--;
graph[dst].push_back(src);
}
bool* vis = new bool[m];
int32_t* matched = new int32_t[n];
int32_t left = m - 2, right = n + 1;
while(right - left > 1) {
int32_t mid = (left + right) / 2;
int32_t matched_cnt = 0;
for(int32_t i = 0; i < n; i++)
matched[i] = -1;
for(int32_t i = 0; i < m; i++) {
for(int32_t j = 0; j < m; j++)
vis[j] = false;
if(try_kuhn(i, vis, graph, matched, 0, mid))
matched_cnt++;
}
if(matched_cnt == m)
right = mid;
else
left = mid;
}
if(left == n) {
std::cout << 0;
return 0;
}
int32_t matched_cnt = 0;
for(int32_t i = 0; i < n; i++)
matched[i] = -1;
for(int32_t i = 0; i < m; i++) {
for(int32_t j = 0; j < m; j++)
vis[j] = false;
if(try_kuhn(i, vis, graph, matched, 0, right))
matched_cnt++;
}
int32_t right_ptr = right;
int32_t ans = n - right_ptr;
for(int32_t i = 1; i < n; i++) {
int32_t rematched = -1;
if(matched[i - 1] != -1) {
matched_cnt--;
rematched = matched[i - 1];
}
if(rematched != -1) {
while(right_ptr < n - 1 && matched_cnt < m) {
right_ptr++;
for(int32_t j = 0; j < m; j++)
vis[j] = false;
if(try_kuhn(rematched, vis, graph, matched, i, right_ptr))
matched_cnt++;
}
}
if(matched_cnt < m)
break;
ans += n - right_ptr;
}
std::cout << ans << "\n";
return 0;
}
Compilation message
marriage.cpp: In function 'bool try_kuhn(int32_t, bool*, std::vector<int>*, int32_t*, int32_t, int32_t)':
marriage.cpp:26:26: warning: comparison of integer expressions of different signedness: 'int32_t' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
26 | for(int32_t i = 0; i < graph[v].size(); i++)
| ~~^~~~~~~~~~~~~~~~~
marriage.cpp:31:26: warning: comparison of integer expressions of different signedness: 'int32_t' {aka 'int'} and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
31 | for(int32_t i = 0; i < graph[v].size(); i++)
| ~~^~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
204 KB |
Output isn't correct |
2 |
Runtime error |
128 ms |
262148 KB |
Execution killed with signal 9 |
3 |
Runtime error |
117 ms |
262148 KB |
Execution killed with signal 9 |
4 |
Runtime error |
123 ms |
262148 KB |
Execution killed with signal 9 |
5 |
Correct |
0 ms |
204 KB |
Output is correct |
6 |
Correct |
0 ms |
204 KB |
Output is correct |
7 |
Runtime error |
135 ms |
262148 KB |
Execution killed with signal 9 |
8 |
Correct |
0 ms |
312 KB |
Output is correct |
9 |
Runtime error |
120 ms |
262148 KB |
Execution killed with signal 9 |
10 |
Runtime error |
120 ms |
262148 KB |
Execution killed with signal 9 |
11 |
Correct |
0 ms |
204 KB |
Output is correct |
12 |
Correct |
0 ms |
204 KB |
Output is correct |
13 |
Correct |
0 ms |
308 KB |
Output is correct |
14 |
Runtime error |
140 ms |
262148 KB |
Execution killed with signal 9 |
15 |
Runtime error |
126 ms |
262148 KB |
Execution killed with signal 9 |
16 |
Runtime error |
119 ms |
262148 KB |
Execution killed with signal 9 |
17 |
Runtime error |
121 ms |
262148 KB |
Execution killed with signal 9 |
18 |
Runtime error |
119 ms |
262148 KB |
Execution killed with signal 9 |
19 |
Runtime error |
273 ms |
262148 KB |
Execution killed with signal 9 |
20 |
Runtime error |
151 ms |
262148 KB |
Execution killed with signal 9 |
21 |
Runtime error |
122 ms |
262148 KB |
Execution killed with signal 9 |
22 |
Runtime error |
118 ms |
262148 KB |
Execution killed with signal 9 |
23 |
Runtime error |
177 ms |
262148 KB |
Execution killed with signal 9 |
24 |
Runtime error |
170 ms |
262148 KB |
Execution killed with signal 9 |
25 |
Runtime error |
480 ms |
262148 KB |
Execution killed with signal 9 |
26 |
Runtime error |
176 ms |
262148 KB |
Execution killed with signal 9 |
27 |
Runtime error |
120 ms |
262148 KB |
Execution killed with signal 9 |
28 |
Runtime error |
120 ms |
262148 KB |
Execution killed with signal 9 |
29 |
Runtime error |
650 ms |
262148 KB |
Execution killed with signal 9 |
30 |
Runtime error |
558 ms |
262148 KB |
Execution killed with signal 9 |
31 |
Runtime error |
381 ms |
262148 KB |
Execution killed with signal 9 |
32 |
Runtime error |
126 ms |
262148 KB |
Execution killed with signal 9 |
33 |
Runtime error |
114 ms |
262148 KB |
Execution killed with signal 9 |
34 |
Runtime error |
120 ms |
262148 KB |
Execution killed with signal 9 |
35 |
Runtime error |
781 ms |
262148 KB |
Execution killed with signal 9 |
36 |
Runtime error |
675 ms |
262148 KB |
Execution killed with signal 9 |
37 |
Runtime error |
234 ms |
262148 KB |
Execution killed with signal 9 |
38 |
Runtime error |
610 ms |
262148 KB |
Execution killed with signal 9 |
39 |
Runtime error |
122 ms |
262148 KB |
Execution killed with signal 9 |
40 |
Runtime error |
129 ms |
262148 KB |
Execution killed with signal 9 |
41 |
Runtime error |
136 ms |
262148 KB |
Execution killed with signal 9 |
42 |
Incorrect |
6 ms |
792 KB |
Output isn't correct |
43 |
Incorrect |
8 ms |
1080 KB |
Output isn't correct |
44 |
Incorrect |
21 ms |
1732 KB |
Output isn't correct |
45 |
Incorrect |
9 ms |
996 KB |
Output isn't correct |
46 |
Runtime error |
297 ms |
262148 KB |
Execution killed with signal 9 |
47 |
Incorrect |
20 ms |
1940 KB |
Output isn't correct |
48 |
Incorrect |
16 ms |
1860 KB |
Output isn't correct |
49 |
Runtime error |
229 ms |
262148 KB |
Execution killed with signal 9 |
50 |
Runtime error |
119 ms |
262148 KB |
Execution killed with signal 9 |