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