C++ の動作確認をしてみた（８７０）

C++の練習を兼ねて, AtCoder Grand Contest 030 の問題C (Coloring Torus) を解いてみた.

■感想.
1. 問題C は, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 個人的には, 解説のロジックで, K色でのよい塗り方を構成ことが, 非常に不思議な印象を受け, 面白く感じた.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.

本家のサイト AtCoder Grand Contest 030 解説をご覧下さい.

■C++版プログラム(問題C／AC版).

// 解き直し.
// https://img.atcoder.jp/agc030/editorial.pdf
// C++(GCC 9.2.1)
#include <bits/stdc++.h>
using namespace std;
#define repex(i, a, b, c) for(int i = a; i < b; i += c)
#define repx(i, a, b) repex(i, a, b, 1)
#define rep(i, n) repx(i, 0, n)
#define repr(i, a, b) for(int i = a; i >= b; i--)
int ans[505][505];

int main(){
    
    // 1. 入力情報.
    int K;
    scanf("%d", &K);
    
    // 2. 例外.
    if(K == 1){
        puts("1\n1");
        return 0;
    }
    
    // 3. K色でのよい塗り方(解説通り).
    int n = 2 * ((K + 3) / 4);
    int d = 2 * n - K;
    rep(r, n){
        rep(c, n){
            int b = (r + c) % n;
            ans[r][c] = b;
            if(r & 1) ans[r][c] += (b < n - d) ? n : 0;
        }
    }
    
    // 4. 出力.
    printf("%d\n", n);
    rep(r, n) rep(c, n) printf("%d%s", ans[r][c] + 1, (c < n - 1) ? " " : "\n");
    return 0;
    
}

// 解き直し.

// https://img.atcoder.jp/agc030/editorial.pdf

// C++(GCC 9.2.1)

#include <bits/stdc++.h>

using namespace std;

#define repex(i, a, b, c) for(int i = a; i < b; i += c)

#define repx(i, a, b) repex(i, a, b, 1)

#define rep(i, n) repx(i, 0, n)

#define repr(i, a, b) for(int i = a; i >= b; i--)

int ans[505][505];

int main(){

// 1. 入力情報.

int K;

scanf("%d", &K);

// 2. 例外.

if(K == 1){

puts("1\n1");

return 0;

}

// 3. K色でのよい塗り方(解説通り).

int n = 2 * ((K + 3) / 4);

int d = 2 * n - K;

rep(r, n){

rep(c, n){

int b = (r + c) % n;

ans[r][c] = b;

if(r & 1) ans[r][c] += (b < n - d) ? n : 0;

}

// 4. 出力.

printf("%d\n", n);

rep(r, n) rep(c, n) printf("%d%s", ans[r][c] + 1, (c < n - 1) ? " " : "\n");

return 0;

}

[入力例]
2

[出力例]
3
1 1 1
1 1 1
2 2 2
※AtCoderテストケースより

※但し, 上記のプログラムでは, 以下の内容が出力される.
2
1 2
2 1

[入力例]
9

[出力例]
3
1 2 3
4 5 6
7 8 9
※AtCoderテストケースより

※但し, 上記のプログラムでは, 以下の内容が出力される.
6
1 2 3 4 5 6
8 9 4 5 6 7
3 4 5 6 1 2
4 5 6 7 8 9
5 6 1 2 3 4
6 7 8 9 4 5

[入力例]
1

[出力例]
1
1

[入力例]
3

[出力例]
2
1 2
2 3

[入力例]
4

[出力例]
2
1 2
4 3

[入力例]
13

[出力例]
8
1 2 3 4 5 6 7 8
10 11 12 13 6 7 8 9
3 4 5 6 7 8 1 2
12 13 6 7 8 9 10 11
5 6 7 8 1 2 3 4
6 7 8 9 10 11 12 13
7 8 1 2 3 4 5 6
8 9 10 11 12 13 6 7

[入力例]
22

[出力例]
12
1 2 3 4 5 6 7 8 9 10 11 12
14 15 16 17 18 19 20 21 22 11 12 13
3 4 5 6 7 8 9 10 11 12 1 2
16 17 18 19 20 21 22 11 12 13 14 15
5 6 7 8 9 10 11 12 1 2 3 4
18 19 20 21 22 11 12 13 14 15 16 17
7 8 9 10 11 12 1 2 3 4 5 6
20 21 22 11 12 13 14 15 16 17 18 19
9 10 11 12 1 2 3 4 5 6 7 8
22 11 12 13 14 15 16 17 18 19 20 21
11 12 1 2 3 4 5 6 7 8 9 10
12 13 14 15 16 17 18 19 20 21 22 11

[入力例]
50

[出力例]
26
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2
30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33
9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10
38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37
13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12
40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14
42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43
19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

[入力例]

[出力例]

1 1 1

2 2 2

※AtCoderテストケースより

※但し, 上記のプログラムでは, 以下の内容が出力される.

1 2

2 1

[入力例]

[出力例]

1 2 3

4 5 6

7 8 9

※AtCoderテストケースより

※但し, 上記のプログラムでは, 以下の内容が出力される.

1 2 3 4 5 6

8 9 4 5 6 7

3 4 5 6 1 2

4 5 6 7 8 9

5 6 1 2 3 4

6 7 8 9 4 5

[入力例]

[出力例]

[入力例]

[出力例]

1 2

2 3

[入力例]

[出力例]

1 2

4 3

[入力例]

[出力例]

1 2 3 4 5 6 7 8

10 11 12 13 6 7 8 9

3 4 5 6 7 8 1 2

12 13 6 7 8 9 10 11

5 6 7 8 1 2 3 4

6 7 8 9 10 11 12 13

7 8 1 2 3 4 5 6

8 9 10 11 12 13 6 7

[入力例]

[出力例]

1 2 3 4 5 6 7 8 9 10 11 12

14 15 16 17 18 19 20 21 22 11 12 13

3 4 5 6 7 8 9 10 11 12 1 2

16 17 18 19 20 21 22 11 12 13 14 15

5 6 7 8 9 10 11 12 1 2 3 4

18 19 20 21 22 11 12 13 14 15 16 17

7 8 9 10 11 12 1 2 3 4 5 6

20 21 22 11 12 13 14 15 16 17 18 19

9 10 11 12 1 2 3 4 5 6 7 8

22 11 12 13 14 15 16 17 18 19 20 21

11 12 1 2 3 4 5 6 7 8 9 10

12 13 14 15 16 17 18 19 20 21 22 11

[入力例]

[出力例]

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26

28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27

3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2

30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29

5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4

32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31

7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6

34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33

9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8

36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35

11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10

38 39 40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37

13 14 15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12

40 41 42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39

15 16 17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14

42 43 44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41

17 18 19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

44 45 46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43

19 20 21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

46 47 48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45

21 22 23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20

48 49 50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47

23 24 25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22

50 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49

25 26 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24

26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 25

■参照サイト
AtCoder Grand Contest 030

C++の練習を兼ねて, AtCoder Grand Contest 030 の 問題C (Coloring Torus) を解いてみた.

コメントを残す コメントをキャンセル

C++の練習を兼ねて, AtCoder Grand Contest 030 の問題C (Coloring Torus) を解いてみた.

コメントを残すコメントをキャンセル