宿舍里的故事之五子棋

宿舍里的故事之五子棋

描述
宿舍里好多好多有趣的事！
7890653今天看到不知何时流行的五子棋，在宿舍里拿个本子，画一些格子，一个棋盘就做好了！
当7890653把目光放到棋上，突发奇想，呵呵！一个题目就出来了！
在一个 5*5 的棋盘内，放上n颗棋子，其中（5<=n<=25)；
这n颗棋子可以不同的放到任何一个地方—在棋盘内！于是呼，便会有五颗棋子排成一行，
或一列，或两条对角线，不同的放法，就会出现多少排五子的排列！
本题你要做的是，给你一个n，你找出不同放法出现的排列(设为k)，如：
n=11;
有(1是棋子，0是空格）
1 1 1 0 0
1 1 0 0 0
1 1 0 0 0
1 1 0 0 0
1 1 0 0 0
k=2;
1 1 1 1 1
1 1 1 1 0
1 1 0 0 0
0 0 0 0 0
0 0 0 0 0
k=1;
只有这两种k值，（注意k不重复），你要输出的便是k值的和。
也就是1+2=3！！！！！！

---

由于时间限制，将5*5的棋盘转化成一维，为什么二维的超时了（为什么？）？

---

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int n,b[6][6],ans[50];
void dfs(int step,int Idx) {
    if(step == 0) {
        int sum = 0;
        for(int i = 1; i<=5; i++) {
            if(!b[i][i]) {
                break;
            }
            if(i == 5) {
                sum++;
            }
        }

        for(int i = 1; i<=5; i++) {
            if(!b[i][6-i]) {
                break;
            }
            if(i == 5) {
                sum++;
            }
        }

        for(int i = 1; i<=5; i++) {
            for(int j = 1; j<=5; j++) {
                if(!b[i][j]) {
                    break;
                }
                if(j == 5) {
                    sum++;
                }
            }
            for(int j = 1; j<=5; j++) {
                if(!b[j][i]) {
                    break;
                }
                if(j == 5) {
                    sum++;
                }
            }
        }
        ans[sum]++;
        return;
    }
    for(int i = Idx; i<=25; i++) {
        int x = (i - 1) / 5 + 1;
        int y = (i - 1) % 5 + 1;
        if(!b[x][y]){
            b[x][y] = 1;
            dfs(step-1,i+1);
            b[x][y] = 0;
        }
    }
}
int main() {
    cin>>n;
    dfs(n,1);
    int re = 0;
    for(int i = 1; i<=36; i++) {
        if(ans[i]) {
            re += i;
        }
    }
    cout<<re;
    return 0;
}