usaco/transform.cpp at master · cadaniluk/usaco · GitHub

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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
/*
ID: th3c0r11
TASK: transform
LANG: C++14
*/

#include <fstream>
#include <iostream>
#include <algorithm>
#include <vector>

int n;

std::vector<char> deg90(const std::vector<char> &a)
{
    std::vector<char> ret;
    int x = 0, y = n - 1;
    for (int i = 0; i < a.size(); ++i) {
        ret.push_back(a[y * n + x]);
        if (y == 0) {
            ++x;
            y = n - 1;
        } else
            --y;
    }
    return ret;
}

std::vector<char> reflect(const std::vector<char> &a)
{
    std::vector<char> ret;

    for (int i = 0; i < n; ++i) {
        for (int j = n - 1; j >= 0; --j) {
            ret.push_back(a[i * n + j]);
        }
    }

    return ret;
}

int main()
{
    std::ifstream in{"transform.in"};
    std::ofstream out{"transform.out"};

    in >> n;
    in.ignore(100, '\n');

    std::vector<char> pattern;

    for (int i = 0; i < n; ++i) {
        std::string str;
        std::getline(in, str);
        for (char c : str)
            pattern.push_back(c);
    }

    std::vector<char> result;

    for (int i = 0; i < n; ++i) {
        std::string str;
        std::getline(in, str);
        for (char c : str)
            result.push_back(c);
    }

    if (deg90(pattern) == result)
        out << "1\n";
    else if (deg90(deg90(pattern)) == result)
        out << "2\n";
    else if (deg90(deg90(deg90(pattern))) == result)
        out << "3\n";
    else if (reflect(pattern) == result)
        out << "4\n";
    else if (deg90(reflect(pattern)) == result || deg90(deg90(reflect(pattern))) == result || deg90(deg90(deg90(reflect(pattern)))) == result)
        out << "5\n";
    else if (pattern == result)
        out << "6\n";
    else
        out << "7\n";
}