USACO 1.5 Checker Challenge (DFS)

#include <stdio.h>
#define DEBUG 1
#define TESTCASES 8
#define IN(x)  ( (x) >= 1 && (x) <= size ? 1 : 0 )
#define WITHIN(x, y) ( IN(x) && IN(y) ? 1 : 0 )
int size;
int conflict[14][14];
int solution[14];
int numOfSolutions;
int directionArray[4][2] = {{1, 1}, {-1, -1}, {-1, 1}, {1, -1}};

void printConflict(){
	int i, j;
	for (i = 1; i <= size; i++)
		for (j = 1; j <= size; j++)
			printf("%d%c", conflict[i][j], j == size ? '\n' : ' ');
	printf("\n");
}

void setConflict(int row, int column, int flag){
	int i;
 	for (i = 1; i <= size; i++)
		conflict[i][column] += flag;
	int directionIndex;
	for (directionIndex = 0; directionIndex < 4; directionIndex++){
		int deltaX = directionArray[directionIndex][0];
		int deltaY = directionArray[directionIndex][1];
		int x = row + deltaX;
		int y = column + deltaY;
		while ( WITHIN(x, y) ){
			conflict[x][y] += flag;
			x += deltaX;
		    y += deltaY;
		}
	}
}

void dfs(int row){
	//printConflict();
	if (row > size){
		if (++numOfSolutions < 4){
			int i;
			for (i = 1; i <= size; i++)
				printf("%d%c", solution[i], i == size ? '\n' : ' ');
		}
		return;
	}
		
	int column;
	for (column = 1; column <= size; column++){
		if (conflict[row][column] == 0){
			setConflict(row, column, 1);
			solution[row] = column;
			dfs(row + 1);
			setConflict(row, column, -1);
		}
	}
}

int main(){
#if DEBUG
	int testCase;
	for (testCase = 1; testCase <= TESTCASES; testCase++){
		char inputFileName[20] = "inputx.txt";
		inputFileName[5] = '1' +  (testCase - 1);
		freopen(inputFileName, "r", stdin);
		printf("\n#%d\n", testCase);
#endif

	scanf("%d", &size);

	numOfSolutions = 0;
	dfs(1);
	printf("%d\n", numOfSolutions);

#if DEBUG
	}
#endif
	return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值