#include <bits/stdc++.h>
using namespace std;
const int MAXN = 500005;
int N, K;
int A[MAXN], B[MAXN];
int ed[MAXN];
vector<int> chi[MAXN];
bool chk[MAXN], col[MAXN];
vector<int> cyc;
int nex[2 * MAXN];
int ans;
int dfs(int x) {
//cout << "dfs(" << x << ")\n";
chk[x] = true;
int go = 0;
for(auto a : chi[x]) go = max(go, dfs(a));
if(x == 1) go = K - 1;
else if(go-- == 0) {
ans++;
go = K - 1;
}
return go;
}
void searchgraph(int x) {
//cout << "searchgraph(" << x << ")\n";
for(; !chk[x]; x = ed[x]) chk[x] = true;
for(cyc.clear(); cyc.empty() || cyc[0] != x; x = ed[x]) cyc.push_back(x);
int go = 0;
for(int i = 0; i < cyc.size(); i++) {
for(auto a : chi[cyc[i]]) if(a != cyc[(i + cyc.size() - 1) % cyc.size()]) go = max(go, dfs(a));
//cout << "go = " << go << "\n";
if(go-- > 0) col[i] = true;
else col[i] = false;
}
for(int i = 0; i < go; i++) col[i] = true;
bool done = true;
for(int i = 0; i < cyc.size(); i++) done &= col[i];
/*for(int i = 0; i < cyc.size(); i++) cout << cyc[i] << " ";
cout << "\n";
for(int i = 0; i < cyc.size(); i++) cout << int(col[i]) << " ";
cout << "\n";
cout << int(done) << "\n";*/
if(done) return;
if(cyc.size() <= K) {
ans++;
return;
}
for(nex[0] = 0; col[nex[0] % cyc.size()]; nex[0]++);
for(int i = 1; i < cyc.size(); i++) {
nex[i] = nex[i - 1] - 1;
if(nex[i] < 0) for(nex[i]++; col[(i + nex[i]) % cyc.size()]; nex[i]++);
}
//for(int i = 0; i < cyc.size(); i++) cout << nex[i] << " ";
//cout << "\n";
//cout << "ans = " << ans << "\n";
int mn = cyc.size();
for(int i = 0; i < K; i++) {
int cnt = 0;
for(int t = i + nex[i]; t < cyc.size() + i; t += K + nex[(t + K) % cyc.size()]) cnt++;
//cout << cnt << " ";
mn = min(mn, cnt);
}
//cout << "\n";
ans += mn;
}
int main() {
ios::sync_with_stdio(0); cin.tie(0);
cin >> N >> K;
for(int i = 0; i < N; i++) cin >> A[i] >> B[i];
for(int i = 0; i < N; i++) {
ed[A[i]] = B[i];
chi[B[i]].push_back(A[i]);
}
for(int i = 1; i <= N; i++) if(!chk[i]) searchgraph(i);
cout << ans;
return 0;
}
Compilation message
link.cpp: In function 'void searchgraph(int)':
link.cpp:33:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i = 0; i < cyc.size(); i++) {
~~^~~~~~~~~~~~
link.cpp:41:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i = 0; i < cyc.size(); i++) done &= col[i];
~~^~~~~~~~~~~~
link.cpp:48:16: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if(cyc.size() <= K) {
~~~~~~~~~~~^~~~
link.cpp:53:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i = 1; i < cyc.size(); i++) {
~~^~~~~~~~~~~~
link.cpp:63:29: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int t = i + nex[i]; t < cyc.size() + i; t += K + nex[(t + K) % cyc.size()]) cnt++;
~~^~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
13 ms |
12152 KB |
Output is correct |
2 |
Incorrect |
13 ms |
12152 KB |
Output isn't correct |
3 |
Correct |
12 ms |
12152 KB |
Output is correct |
4 |
Correct |
13 ms |
12280 KB |
Output is correct |
5 |
Incorrect |
15 ms |
12280 KB |
Output isn't correct |
6 |
Correct |
25 ms |
13532 KB |
Output is correct |
7 |
Correct |
35 ms |
14456 KB |
Output is correct |
8 |
Incorrect |
42 ms |
15580 KB |
Output isn't correct |
9 |
Correct |
60 ms |
18576 KB |
Output is correct |
10 |
Correct |
62 ms |
17024 KB |
Output is correct |
11 |
Correct |
95 ms |
20904 KB |
Output is correct |
12 |
Correct |
141 ms |
21228 KB |
Output is correct |
13 |
Incorrect |
185 ms |
25868 KB |
Output isn't correct |
14 |
Correct |
223 ms |
25976 KB |
Output is correct |
15 |
Correct |
260 ms |
29448 KB |
Output is correct |
16 |
Correct |
316 ms |
30764 KB |
Output is correct |
17 |
Correct |
341 ms |
33664 KB |
Output is correct |
18 |
Incorrect |
401 ms |
34652 KB |
Output isn't correct |
19 |
Correct |
393 ms |
35188 KB |
Output is correct |
20 |
Correct |
399 ms |
36212 KB |
Output is correct |
21 |
Incorrect |
402 ms |
37620 KB |
Output isn't correct |
22 |
Correct |
403 ms |
36632 KB |
Output is correct |
23 |
Correct |
409 ms |
36260 KB |
Output is correct |
24 |
Correct |
397 ms |
36420 KB |
Output is correct |
25 |
Correct |
402 ms |
36088 KB |
Output is correct |