宿舍裏的故事之五子棋

宿舍裏的故事之五子棋

描述
宿舍裏好多好多有趣的事！
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;
}