【DFS】蓝桥杯试题 兰顿蚂蚁

题目描述

OJ地址 http://lx.lanqiao.cn/problem.page?gpid=T125

　　 兰顿蚂蚁，是于1986年，由克里斯·兰顿提出来的，属于细胞自动机的一种。

平面上的正方形格子被填上黑色或白色。在其中一格正方形内有一只“蚂蚁”。
　　蚂蚁的头部朝向为：上下左右其中一方。

蚂蚁的移动规则十分简单：
　　若蚂蚁在黑格，右转90度，将该格改为白格，并向前移一格；
　　若蚂蚁在白格，左转90度，将该格改为黑格，并向前移一格。

规则虽然简单，蚂蚁的行为却十分复杂。刚刚开始时留下的路线都会有接近对称，像是会重复，但不论起始状态如何，蚂蚁经过漫长的混乱活动后，会开辟出一条规则的“高速公路”。

蚂蚁的路线是很难事先预测的。

你的任务是根据初始状态，用计算机模拟兰顿蚂蚁在第n步行走后所处的位置。

输入：

输入数据的第一行是 m n 两个整数（3 < m, n < 100），表示正方形格子的行数和列数。
　　接下来是 m 行数据。
　　每行数据为 n 个被空格分开的数字。0 表示白格，1 表示黑格。

接下来是一行数据：x y s k, 其中x y为整数，表示蚂蚁所在行号和列号（行号从上到下增长，列号从左到右增长，都是从0开始编号）。s 是一个大写字母，表示蚂蚁头的朝向，我们约定：上下左右分别用：UDLR表示。k 表示蚂蚁走的步数。

输出：

输出数据为两个空格分开的整数 p q, 分别表示蚂蚁在k步后，所处格子的行号和列号。

样例输入1：

5 6
0 0 0 0 0 0
0 0 0 0 0 0
0 0 1 0 0 0
0 0 0 0 0 0
0 0 0 0 0 0
2 3 L 5

样例输出1：

1 3

样例输入2：

3 3
0 0 0
1 1 1
1 1 1
1 1 U 6

样例输出2：

0 0

思路：

不复杂，直接dfs。注意一下头的朝向，走法就可以了。

代码

#include <bits/stdc++.h>

using namespace std;


int r,c,mp[105][105]; 
int k;
map<char,pair<int,int> > dir_left;
map<char,pair<int,int> > dir_right;

void init(){
	dir_left['U'] = make_pair(0,-1);
	dir_left['D'] = make_pair(0,1);
	dir_left['L'] = make_pair(1,0);
	dir_left['R'] = make_pair(-1,0);
	
	dir_right['U'] = make_pair(0,1);
	dir_right['D'] = make_pair(0,-1);
	dir_right['L'] = make_pair(-1,0);
	dir_right['R'] = make_pair(1,0);
	
}


void dfs(int x,int y, char s, int cnt){
	
	
	if(cnt == k){
		cout << x << " " << y << endl;
		return;
	} 
	
	if(mp[x][y] == 0){  // 白  左转  
		mp[x][y] = 1;
		// 开始走下一步
		int dx = x + dir_left[s].first;
		int dy = y + dir_left[s].second;
		char ds;
		
		switch(s){
			case 'U': ds = 'L'; break;
			case 'D': ds = 'R'; break;
			case 'L': ds = 'D'; break;
			case 'R': ds = 'U'; break;
		} 
		dfs(dx,dy,ds,cnt+1);
	}else{
		// 黑 右转
		mp[x][y] = 0;
		int dx = x + dir_right[s].first;
		int dy = y + dir_right[s].second;
		char ds ;
		switch(s){
			case 'U': ds = 'R'; break;
			case 'D': ds = 'L'; break;
			case 'L': ds = 'U'; break;
			case 'R': ds = 'D'; break;
		} 
		dfs(dx,dy,ds,cnt+1);
	}
	
	
}

int main(){
	init();
	char s;
	int x,y;
	cin >> r >> c;
	for(int i = 0 ; i < r; i++){
		for(int j = 0; j < c; j++){
			cin >> mp[i][j];
		}
	}
	cin >> x >> y >> s >> k;

	dfs(x,y,s,0);

	return 0;
}