Project Euler 18

http://projecteuler.net/index.php?section=problems&id=18

整数をルートに割り当てる。次に左右どちらに行くかを2進数で表して各桁が0か1かで決める。それを関数化するとiterateでルートが作れる。

itertools.h

#include <iostream>
#include "itertools.h"

using namespace std;
using namespace itertools;

char    *s[] = {
        "75",
        "95 64",
        "17 47 82",
        "18 35 87 10",
        "20 04 82 47 65",
        "19 01 23 75 03 34",
        "88 02 77 73 07 63 67",
        "99 65 04 28 06 16 70 92",
        "41 41 26 56 83 40 80 70 33",
        "41 48 72 33 47 32 37 16 94 29",
        "53 71 44 65 25 43 91 52 97 51 14",
        "70 11 33 28 77 73 17 78 39 68 17 57",
        "91 71 52 38 17 14 91 43 58 50 27 29 48",
        "63 66 04 68 89 53 67 30 73 16 69 87 40 31",
        "04 62 98 27 23 09 70 98 73 93 38 53 60 04 23"
};
const int   size = sizeof(s) / sizeof(char *);
vector<vector<int>> v;

tuple<int,int,int> next(tuple<int,int,int> x) {
    int n = get<2>(x);
    int dx = n & 1;
    return make_tuple(get<0>(x) + dx, get<1>(x) + 1, n >> 1);
}

int sum_path(int n) {
    return sum(map(
            [] (tuple<int,int,int>x) { return v[get<1>(x)][get<0>(x)]; },
            take(size, iterate(next, make_tuple(0, 0, n)))));
}

int main() {
    for(int k = 0; k < size; k++)
        v.push_back(list(map(
                [] (string& s) { return stoi(s); }, words(s[k]))));
    cout << reduce1(max<int>, map(sum_path, range<>(1 << size))) << endl;
}