blob: 38d602b31128513d553c56e9b68320ae91ee0c33 (
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
|
#include <iostream>
#include <fstream>
#include <string>
#include <vector>
#include "../utils.h"
long sum (std::vector<long> vec) {
long s = 0; for (long v : vec) s += v; return s;
}
int main (int argc, char *argv[]) {
std::string raw;
std::getline(std::ifstream(argv[1]), raw, '\0');
std::vector<std::string> strfish;
split(strfish, raw, ",");
std::vector<long> fish;
for (int i=0; i<9; i++) { fish.push_back(0); }
for (std::string str : strfish) fish[std::stoi(str)] += 1;
int days = 80;
if (argc > 2) days = std::stoi(std::string(argv[2]));
for (int i=0; i<days; i++) {
long born = fish[0];
fish[0] = 0;
for (int j=1; j<9; j++) {
fish[j-1] += fish[j];
fish[j] = 0;
}
fish[6] += born;
fish[8] = born;
if (argc > 3 && std::string(argv[3]) == "-w") {
std::cout << "day " << i+1 << "/" << days
<< " current fish: " << sum(fish) << std::endl;
}
if (argc > 3 && std::string(argv[3]) == "-v") {
std::cout << "day " << i+1 << ": ";
for (auto f : fish) std::cout << f << ",";
std::cout << std::endl;
}
}
std::cout << "fish: " << sum(fish) << std::endl;
return 0;
}
|
