eight_princess.cpp

来自「超强的八后算法」· C++ 代码 · 共 37 行

#include <stdio.h>
#include <stdlib.h>
#include <time.h>

long sum = 0, upperlim = 1;

void test(long row, long ld, long rd){
        if (row != upperlim){
                long pos = upperlim & ~(row | ld | rd);
                while (pos){
                        long p = pos& - pos;
                        pos -= p;
                        test(row + p, (ld + p) << 1, (rd + p) >> 1);
                }
        }
        else
                sum++;
}

void main(int argc, char *argv[]){
        time_t tm;
        int n = 15;

        if(argc != 1) n = atoi(argv[1]);
        tm = time(0);
        if((n < 1) || (n > 32)){
                printf(" heh..I can't calculate that.\n");
        exit(-1);
        }
        printf("%d Queens\n", n);
        upperlim = (upperlim << n) - 1;

        test(0, 0, 0);
        printf("Number of solutions is %ld, %d seconds\n", sum, (int)(time(0) - tm));
}