aboutsummaryrefslogtreecommitdiff
path: root/2017/src/day10.cpp
blob: d300a065544b521284c0a04a6f7b6b65379822aa (plain)
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#include <filesystem>
#include <fstream>
#include <iostream>
#include <numeric>
#include <sstream>
#include <string>
#include <vector>

using namespace std;

namespace {

string
read_file(const filesystem::path& filename)
{
	ifstream file{ filename };
	string   line;
	getline(file, line);
	return line;
}

vector<size_t>
split(const string& line)
{
	stringstream   strm{ line };
	vector<size_t> data;

	for ( string word; getline(strm, word, ','); ) {
		data.emplace_back(stoul(word));
	}

	return data;
}

vector<unsigned>
gen_data(size_t len)
{
	vector<unsigned> data(len);
	iota(data.begin(), data.end(), 0UL);
	return data;
}

void
reverse(vector<unsigned>& data, size_t start, size_t len)
{
	auto end = start + len;

	while ( start < end ) {
		swap(data.at(start++ % data.size()), data.at(--end % data.size()));
	}
}

void
hash(vector<unsigned>& data, const vector<size_t>& lengths)
{
	size_t pos = 0;
	for ( size_t skip = 0; skip != lengths.size(); ++skip ) {
		reverse(data, pos, lengths.at(skip));
		pos += lengths.at(skip) + skip;
	}
}

void
part1(size_t len, const string& line)
{
	auto data    = gen_data(len);
	auto lengths = split(line);

	hash(data, lengths);

	cout << "Part1: " << data.at(0) * data.at(1) << '\n';
}

void
hash64(vector<unsigned>& data, const vector<size_t>& lengths)
{
	size_t pos  = 0;
	size_t skip = 0;
	for ( size_t i = 0; i != 64; ++i ) {
		for ( auto length: lengths ) {
			reverse(data, pos, length);
			pos += length + skip;
			++skip;
		}
	}
}

void
part2(string line)
{
	line += char(17);
	line += char(31);
	line += char(73);
	line += char(47);
	line += char(23);

	vector<size_t> lengths;
	for ( const auto byte: line ) {
		lengths.emplace_back(static_cast<unsigned char>(byte));
	}

	auto data = gen_data(256);
	hash64(data, lengths);

	cout << "Part2: ";
	for ( size_t i = 0; i != 256; i += 16 ) {
		size_t value = 0;
		for ( size_t j = 0; j != 16; ++j ) {
			value ^= data.at(i + j);
		}
		cout << setw(2) << setfill('0') << hex << (value & 0xFF);
	}
	cout << '\n';
}

} // namespace

int
main()
{
	auto line = read_file("data/day10.txt");
	part1(256, line);
	part2(line);
}