QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#610450#9412. Stop the CastleUESTC_OldEastWestAC ✓15ms3840kbC++177.3kb2024-10-04 16:00:532024-10-04 16:00:53

Judging History

This is the latest submission verdict.

  • [2024-10-04 16:00:53]
  • Judged
  • Verdict: AC
  • Time: 15ms
  • Memory: 3840kb
  • [2024-10-04 16:00:53]
  • Submitted

answer

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5;

namespace Dinic {
	int n, s, t, tot;
	std::vector<int> to, nxt, cap, head;
	std::vector<int> cur, dep;

	void AddEdge(int u, int v, int c) {
		to[tot] = v, nxt[tot] = head[u], cap[tot] = c, head[u] = tot++;
	}

	void Init(int _n, int _s, int _t, std::vector<std::array<int, 3> > &edge) {
		n = _n, s = _s, t = _t, tot = 0;
		int m = edge.size();
		to = nxt = cap = std::vector<int>(m << 1);
		head = cur = dep = std::vector<int>(n, -1);
		for (auto [u, v, c] : edge) {
			AddEdge(u, v, c);
			AddEdge(v, u, 0);
		}
	}

	bool BFS() {
		dep = std::vector<int>(n, -1);
		std::vector<int> que(1, s);
		dep[s] = 0;
		for (int h = 0, u; h < que.size(); ++h) {
			u = que[h];
			for (int i = head[u], v, c; ~i; i = nxt[i]) {
				v = to[i], c = cap[i];
				if (dep[v] == -1 && c) dep[v] = dep[u] + 1, que.emplace_back(v);
			}
		}
		return dep[t] != -1;
	}

	int DFS (int u, int in) {
		if (u == t || !in) return in;
		int out = 0;
		for (int i = cur[u], v, c; ~i; i = nxt[i]) {
			cur[u] = nxt[i];
			v = to[i], c = cap[i];
			if (c && dep[v] == dep[u] + 1) {
				int ret = DFS(v, std::min(in, c));
				in -= ret, out += ret;
				cap[i] -= ret, cap[i ^ 1] += ret;
			}
		}
		return out;
	}

	int getMaxFlow() {
		int ans = 0;
		while (BFS()) {
			cur = head;
			ans += DFS(s, INT_MAX);
		}
		return ans;
	}
}


void charming() {
  int n; std::cin >> n;
  std::vector<std::pair<int, int> > castle;
  for (int i = 0, r, c; i < n; ++i) {
    std::cin >> r >> c;
    castle.emplace_back(std::make_pair(r, c));
  }

  int m; std::cin >> m;
  std::vector<std::pair<int, int> > obstacle;
  for (int i = 0, r, c; i < m; ++i) {
    std::cin >> r >> c;
    obstacle.emplace_back(std::make_pair(r, c));
  }

  std::function<bool(int, int, int)> Check = [&](int x, int y, int z) {
    if (x > y) std::swap(x, y);
    if (z > x && z < y) return true;
    else return false;
  };

  std::function<bool(int, int, int, int)> hinder = [&](int idx0, int idx1, int idx2, int type) {
    bool r_same = castle[idx0].first == castle[idx1].first;
    if (type == 0) {
      if (r_same) {
      	if (castle[idx2].first != castle[idx0].first) return false;
        int c0 = castle[idx0].second, c1 = castle[idx1].second, c2 = castle[idx2].second;
        return Check(c0, c1, c2);
      }
      else {
      	if (castle[idx2].second != castle[idx0].second) return false;
        int r0 = castle[idx0].first, r1 = castle[idx1].first, r2 = castle[idx2].first;
        return Check(r0, r1, r2);
      }
    }
    else {
      if (r_same) {
      	if (obstacle[idx2].first != castle[idx0].first) return false;
        int c0 = castle[idx0].second, c1 = castle[idx1].second, c2 = obstacle[idx2].second;
        return Check(c0, c1, c2);
      }
      else {
      	if (obstacle[idx2].second != castle[idx0].second) return false;
        int r0 = castle[idx0].first, r1 = castle[idx1].first, r2 = obstacle[idx2].first;
        return Check(r0, r1, r2);
      }
    }
  };

  std::function<bool(int, int)> fight = [&](int idx0, int idx1) {
    if (castle[idx0].first == castle[idx1].first || 
    castle[idx0].second == castle[idx1].second) {
      for (int i = 0; i < n; ++i) if (i != idx0 && i != idx1) {
        if (hinder(idx0, idx1, i, 0)) return false;
      }
      for (int i = 0; i < m; ++i) {
        if (hinder(idx0, idx1, i, 1)) return false;
      }
      return true;
    }
    else return false;
  };

	bool near = false;
  std::vector<std::pair<int, int> > conf_row, conf_col;
  for (int i = 0; i < n; ++i) {
    for (int j = i + 1; j < n; ++j) {
    	if (abs(castle[i].first - castle[j].first) + 
			abs(castle[i].second - castle[j].second) <= 1) {
    		near = true; break;
    	}
      
			if (castle[i].first == castle[j].first) {
        if (fight(i, j)) conf_row.emplace_back(std::make_pair(i, j));
      }
      else {
        if (fight(i, j)) conf_col.emplace_back(std::make_pair(i, j));
      }
    }
    if (near) break;
  }
  
  if (near) {std::cout << -1 << '\n'; return;}

  int s = 0, t = 1, tot = 2;
  std::map<int, int> mp0, mp1, rev_mp0, rev_mp1;
  std::vector<std::array<int, 3> > edge;
  int conf_row_cnt = conf_row.size(), conf_col_cnt = conf_col.size();
  for (int i = 0; i < conf_row_cnt; ++i) {
    mp0[i] = tot, rev_mp0[tot++] = i;
    edge.emplace_back((std::array<int, 3>) {s, mp0[i], 1});
  }
  for (int i = 0; i < conf_col_cnt; ++i) {
    mp1[i] = tot, rev_mp1[tot++] = i;
    edge.emplace_back((std::array<int, 3>) {mp1[i], t, 1});
  }

  std::function<bool(int, int)> is_inter = [&](int idx0, int idx1) {
    int r0 = conf_row[idx0].first, r1 = conf_row[idx0].second;
    int c0 = conf_col[idx1].first, c1 = conf_col[idx1].second;
    int r = castle[r1].first, c = castle[c1].second;
    if (r <= std::min(castle[c0].first, castle[c1].first)) return false;
    if (r >= std::max(castle[c0].first, castle[c1].first)) return false;
    if (c <= std::min(castle[r0].second, castle[r1].second)) return false;
    if (c >= std::max(castle[r0].second, castle[r1].second)) return false;
    return true;
  };

  std::function<std::pair<int, int>(int, int)> get_inter = [&](int idx0, int idx1) {
    int r = castle[conf_row[idx0].first].first, c = castle[conf_col[idx1].first].second;
    return std::make_pair(r, c);
  };

  for (int i = 0; i < conf_row_cnt; ++i) {
    for (int j = 0; j < conf_col_cnt; ++j) {
      if (is_inter(i, j)) edge.emplace_back((std::array<int, 3>) {mp0[i], mp1[j], 1});
    }
  }

  Dinic::Init(tot, s, t, edge);
  int ans = conf_row_cnt + conf_col_cnt - Dinic::getMaxFlow();

  std::set<std::pair<int, int> > pos;
  {
    using namespace Dinic;
    std::vector<int> vis_r(conf_row_cnt), vis_c(conf_col_cnt);
    for (int u = 0; u < conf_row_cnt; ++u) {
      for (int i = head[mp0[u]], v, c; ~i; i = nxt[i]) if (i & 1 ^ 1) {
        v = rev_mp1[to[i]], c = cap[i];
        if (!c) {
          pos.insert(get_inter(u, v));
          vis_r[u] = 1, vis_c[v] = 1;
        }
      }
    }

    for (int i = 0; i < conf_row_cnt; ++i) if (!vis_r[i]) {
      auto [idx0, idx1] = conf_row[i];
      int r = castle[idx0].first;
      int mn = castle[idx0].second, mx = castle[idx1].second;
      if (mn > mx) std::swap(mn, mx);
      for (int j = mn + 1; j < mx; ++j) {
        std::pair<int, int> new_pos = std::make_pair(r, j);
        if (pos.count(new_pos)) continue;
        else {pos.insert(new_pos); break;}
      }
    }
    for (int i = 0; i < conf_col_cnt; ++i) if (!vis_c[i]) {
      auto [idx0, idx1] = conf_col[i];
      int c = castle[idx0].second;
      int mn = castle[idx0].first, mx = castle[idx1].first;
      if (mn > mx) std::swap(mn, mx);
      pos.insert(std::make_pair(mn + 1, c));
      // for (int j = mn + 1; j < mx; ++j) {
      //   std::pair<int, int> new_pos = std::make_pair(j, c);
      //   if (pos.count(new_pos)) continue;
      //   else {pos.insert(new_pos); break;}
      // }
    }
  }
  
  ans = pos.size();

  std::cout << ans << '\n';
  for (auto [r, c] : pos) std::cout << r << ' ' << c << '\n';
}

signed main() {
  std::ios_base::sync_with_stdio(false);
  std::cin.tie(NULL);
  std::cout.tie(NULL);
  int t; std::cin >> t;
  while (t--) charming();
  return 0;
}

这程序好像有点Bug,我给组数据试试?

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 3596kb

input:

4
7
1 3
6 6
4 7
2 1
6 3
4 1
2 6
3
3 4
6 4
3 1
2
1 1
2 2
0
3
1 1
1 3
3 3
1
1 2
3
1 1
1 3
2 3
0

output:

2
2 3
4 6
0
1
2 3
-1

result:

ok ok 4 cases (4 test cases)

Test #2:

score: 0
Accepted
time: 0ms
memory: 3536kb

input:

21
11
3 5
18 6
19 12
27 48
28 38
30 12
33 18
42 18
43 38
45 46
48 34
3
2 6
24 4
41 45
15
11 6
15 27
16 9
16 14
16 48
19 26
25 12
27 26
28 4
31 40
32 6
33 50
37 50
46 11
50 29
17
1 49
4 9
9 15
9 22
11 31
11 46
14 28
17 5
17 49
18 43
20 31
30 46
33 11
33 33
34 15
36 6
47 10
2
16 46
36 30
11
7 34
7 41
...

output:

3
20 12
29 38
34 18
5
12 6
16 10
16 15
20 26
34 50
0
1
16 10
0
1
43 22
5
1 3
1 13
33 10
42 44
44 45
0
5
8 1
21 15
27 41
29 26
44 4
1
32 9
0
0
0
0
6
12 11
23 10
23 44
24 46
29 21
35 34
0
3
20 30
31 17
43 25
0
-1
3
16 36
17 39
25 7
6
3 10
4 9
5 5
6 4
7 5
8 11

result:

ok ok 21 cases (21 test cases)

Test #3:

score: 0
Accepted
time: 1ms
memory: 3828kb

input:

2
200
7 52
9 160
10 4
12 61
18 436
19 430
20 499
24 139
25 416
29 53
33 153
35 275
35 310
37 25
38 477
39 349
42 120
44 158
45 330
49 486
50 204
51 67
52 138
52 305
56 139
63 132
66 4
67 327
70 484
71 67
72 308
74 218
76 479
81 139
82 90
86 201
87 335
91 35
91 220
92 496
94 29
94 436
96 359
97 299
1...

output:

46
11 4
19 436
21 499
35 276
38 25
39 477
52 67
52 139
57 139
91 160
94 61
126 153
126 214
132 67
138 429
154 138
154 496
185 35
187 433
187 467
199 432
224 147
224 393
240 35
260 275
261 468
270 305
271 186
274 67
277 189
277 356
278 272
280 186
286 367
306 374
311 177
311 455
334 367
349 112
352 6...

result:

ok ok 2 cases (2 test cases)

Test #4:

score: 0
Accepted
time: 1ms
memory: 3624kb

input:

20
13
89287395 441913071
89287395 590314987
142917372 582720391
142917372 598813561
232930851 582720391
263974835 468188689
263974835 490702144
543529670 152471388
543529670 219371706
647691663 598813561
772865038 598813561
773363571 482933265
773363571 561453301
8
128947560 120560639
264980960 4742...

output:

8
89287395 441913072
142917372 582720392
142917373 582720391
142917373 598813561
263974835 468188690
543529670 152471389
647691664 598813561
773363571 482933266
7
59832704 871951216
92745430 358274214
469117951 762966373
808969359 386523246
808969359 818037531
808969359 891229782
808969360 354711322...

result:

ok ok 20 cases (20 test cases)

Test #5:

score: 0
Accepted
time: 2ms
memory: 3660kb

input:

2
184
35763790 435860426
152681166 443483111
261932524 745277126
261932524 787982418
314305867 342102981
314305867 522593781
314305867 747023891
314305867 811404535
314305867 816914009
314305867 857896659
319901983 797458033
321347896 342102981
332550928 902179526
343207116 914408792
350635050 15515...

output:

160
261932524 745277127
314305867 342102982
314305867 522593782
314305867 787982418
314305867 811404536
314305867 816914010
314305868 342102981
314305868 811404535
314305868 857896659
321347897 342102981
332550929 902179526
350635050 155155062
350635050 342102982
350635050 435860426
350635050 526518...

result:

ok ok 2 cases (2 test cases)

Test #6:

score: 0
Accepted
time: 2ms
memory: 3680kb

input:

2
193
78368109 329329995
87932514 657474465
87932514 903492853
94608574 845872655
103602204 88649154
124431127 405860533
145262472 108198774
145262472 907127202
154400439 136367504
165270984 773451933
184157521 409828915
197728547 291882089
197728547 609623645
211750768 843867422
246178069 108396139...

output:

145
87932514 657474466
87932515 903492853
103602205 88649154
145262472 845872655
197728547 405860533
197728548 291882089
246178069 843867422
291242118 564718673
291242119 211732513
365329221 29599938
365329221 62966786
365329221 88649155
365329221 136367504
365329221 141752548
365329221 211655411
36...

result:

ok ok 2 cases (2 test cases)

Test #7:

score: 0
Accepted
time: 2ms
memory: 3612kb

input:

2
192
1075648 579875717
1075648 937648715
1393060 191616432
3957616 541362608
3957616 971879257
5415801 541362608
5887448 541362608
5887448 624842533
25767120 610618164
26193206 214214028
26193206 231445627
26727662 374314271
29550509 755417826
29550509 917592925
30033126 610618164
31134588 58188305...

output:

131
1075648 579875718
3957616 541362609
3957617 541362608
5415802 541362608
5887448 541362609
5887449 541362608
25767121 610618164
26193206 214214029
29550509 755417827
29550510 755417826
31134588 58188306
31134588 231445627
31134589 755417826
33207011 755417826
38104744 231445627
46425938 934726175...

result:

ok ok 2 cases (2 test cases)

Test #8:

score: 0
Accepted
time: 1ms
memory: 3552kb

input:

20
14
378192988 147499872
378192988 837331700
449924898 147499872
449924898 837331700
592684636 147499872
592684636 837331700
783332532 147499872
783332532 837331700
378192988 197547597
783332532 197547597
378192988 343857831
783332532 343857831
378192988 576579017
783332532 576579017
2
449924898 19...

output:

15
378192988 147499873
378192988 197547598
378192988 343857832
378192988 576579018
378192989 147499872
378192989 576579017
378192989 837331700
449924899 147499872
449924899 837331700
592684637 147499872
592684637 837331700
783332532 147499873
783332532 197547598
783332532 343857832
783332532 5765790...

result:

ok ok 20 cases (20 test cases)

Test #9:

score: 0
Accepted
time: 5ms
memory: 3764kb

input:

2
200
8623893 3649468
8623893 989134714
24820918 3649468
24820918 989134714
43705451 3649468
43705451 989134714
45500146 3649468
45500146 989134714
58265855 3649468
58265855 989134714
112263159 3649468
112263159 989134714
126496650 3649468
126496650 989134714
137574156 3649468
137574156 989134714
14...

output:

249
8623893 3649469
8623893 6255308
8623893 11610915
8623893 23366505
8623893 28938722
8623893 31914493
8623893 36029446
8623893 66226936
8623893 84152869
8623893 132710905
8623893 149083020
8623893 174305882
8623893 180413074
8623893 195080122
8623893 231772825
8623893 234663337
8623893 246352595
8...

result:

ok ok 2 cases (2 test cases)

Test #10:

score: 0
Accepted
time: 5ms
memory: 3688kb

input:

2
200
2691939 27674656
2691939 984614319
30225807 27674656
30225807 984614319
39388076 27674656
39388076 984614319
55883389 27674656
55883389 984614319
77258752 27674656
77258752 984614319
106874917 27674656
106874917 984614319
144383292 27674656
144383292 984614319
145360776 27674656
145360776 9846...

output:

216
2691939 27674657
2691939 69029841
2691939 160188719
2691939 176264535
2691939 189367595
2691939 206375959
2691939 234369847
2691939 243902653
2691939 257900117
2691939 268022575
2691939 293316180
2691939 294444587
2691939 363531957
2691939 376231504
2691939 383623457
2691939 386906491
2691939 39...

result:

ok ok 2 cases (2 test cases)

Test #11:

score: 0
Accepted
time: 15ms
memory: 3840kb

input:

2
200
1005937 7
3412691 7
5764908 7
7897785 7
8061992 7
12801501 7
18529696 7
22574605 7
25203351 7
34574958 7
34998216 7
40928451 7
49792193 7
55962320 7
59632864 7
62353373 7
62611196 7
65346591 7
71459320 7
71550121 7
72215989 7
82584263 7
86119549 7
96204862 7
96940713 7
97693112 7
102067050 7
1...

output:

199
1005938 7
3412692 7
5764909 7
7897786 7
8061993 7
12801502 7
18529697 7
22574606 7
25203352 7
34574959 7
34998217 7
40928452 7
49792194 7
55962321 7
59632865 7
62353374 7
62611197 7
65346592 7
71459321 7
71550122 7
72215990 7
82584264 7
86119550 7
96204863 7
96940714 7
97693113 7
102067051 7
105...

result:

ok ok 2 cases (2 test cases)

Extra Test:

score: 0
Extra Test Passed