蓝桥杯 ADV-203 试题 算法提高 8皇后·改

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

　　规则同8皇后问题，但是棋盘上每格都有一个数字，要求八皇后所在格子数字之和最大。

输入格式

　　一个8*8的棋盘。

输出格式

　　所能得到的最大数字和

样例输入

1 2 3 4 5 6 7 8
9 10 11 12 13 14 15 16
17 18 19 20 21 22 23 24
25 26 27 28 29 30 31 32
33 34 35 36 37 38 39 40
41 42 43 44 45 46 47 48
48 50 51 52 53 54 55 56
57 58 59 60 61 62 63 64

样例输出

260

数据规模和约定

　　棋盘上的数字范围0~99

#include <iostream>
#define MAXN 8
using namespace std;
const int num=8;
struct Node{
	int x,y;
};
Node board[MAXN][MAXN];
int count=0;
int MAX=-1;
int sum=0;
bool isRight(int row,int col)
{
	for(int i=0;i<row;i++)
		if(board[i][col].x==1)
			return false;
	for(int i=row,j=col;i>=0&&j>=0;i--,j--)	
		if(board[i][j].x==1)
			return false;
	for(int i=row,j=col;i>=0&&j<num;i--,j++)
		if(board[i][j].x==1)
			return false;
	return true;
}

void dfs(int row)
{
	if(row==num)
	{
		count++;
		for(int i=0;i<num;i++)
		for(int j=0;j<num;j++)
		{
			if(board[i][j].x==1)
				sum+=board[i][j].y; 
			if(sum>MAX)
			{
				MAX=sum;
			}
		} 
		sum=0; 	
		return ;
	}
	else{
		for(int col=0;col<num;col++)
		{
			if(isRight(row,col))
			{
				board[row][col].x=1;
				dfs(row+1);
				board[row][col].x=0;
			}
		}
	}
}
int main()
{
	for(int i=0;i<num;i++)
		for(int j=0;j<num;j++)
		{
			cin>>board[i][j].y;
			board[i][j].x=0;
		}	
	dfs(0);
	printf("%d\n",MAX);
	return 0;
 }