日能研の問題をCで解いてみた(2018/11)

日能研の広告の問題をpythonで解いてみた記事があったので、同じことをC言語で実装してみました。

#include <stdio.h>
#include <math.h>

int trim(int d, int n) {
    int w, k=1, t=0;
    while(d > 0) {
        w = d % 10;
        if(w != n) {
            t += k * w;
            k *= 10;
        }
        d /= 10;
    }
    return t;
}

int ite(int *d)　{
    int r;
    r = *d % 10;
    *d /= 10;
    return r;
}

void output(int val) {
    int i,w;
    w = val;
    for(i=6; i>1; i--) {
        if(w%i != 0)
            break;
        w /= 10;
    }
    if(i <= 1) {
        static int cnt = 1;
        printf("%d\n", val);
    }
}

void permutations(int cl, int k, int val) {
    int t_i, t_cl, t_w;
    if(k-2 <= 0) {
        output(val + cl);
        output(val + (10*(cl%10)) + (cl/10));
        return;
    }
    t_cl = cl;
    while((t_i=ite(&t_cl)) != 0) {
        t_w = t_i * pow(10, k-1);
        permutations(trim(cl,t_i), k-1, val+t_w);
    }
}

int main() {
    permutations(123456, 6, 0);
    return 0;
}

permutations()を再帰的に呼ぶことでpythonのitertools.permutations()と同様のことを行い、生成した値をoutput()に渡して条件を満たす時だけ出力しています。