算法笔记——棋子换位

题目描述

有 nn 个棋子 A，nn 个棋子 B，在棋盘上排成一行。 它们中间隔着一个空位，用“.”表示，比如：

AAA.BBB

现在需要所有的 A 棋子和 B 棋子交换位置。 移动棋子的规则是：

A 棋子只能往右边移动，B 棋子只能往左边移动。
每个棋子可以移动到相邻的空位。
每个棋子可以跳过相异的一个棋子落入空位（A 跳过 B 或者 B 跳过 A ）。
AAA.BBB 可以走法： 移动 A ==> AA.ABBB 移动 B ==> AAAB.BB

跳走的例子： AA.ABBB ==> AABA.BB

分析

可以知道，一个字母的移动，有两种可能：

1. 往前走一步，给另一种棋子作为跳板；
2. 交换结束，只差自己到达正确的位置上。

那么我们考虑一下什么时候移动棋子，会让两个相同字母相邻，从而导致另一种字母无法通过。

当一个棋子需要作为跳板往前移动，此时必然前方是另一种棋子，而身后是和自己同类型的棋子。
当交换结束，只差自己到达正确的位置上，此时必然前方是和自己同类型的棋子，而后方是已经成功交换的另一方棋子。
所以条件判断很明显，即判断当前行动的棋子的前后棋子是否不同即可，若是相同类型则不能移动，continue 掉。细节是需要用 valid 函数判断访问地址是否合法，以防越界。

代码实现

#include<stdio.h>
#include<string.h>

//用于执行移动棋子的步骤，把位于from的棋子走到to的空位置
void move(char* data, int from, int to)
{
    data[to] = data[from];
    data[from] = '.';
}

//判断k是否越界
int valid(char* data, int k)
{
    if(k<0 || k>=strlen(data)) return 0;
    return 1;
}

//函数主体，完成了在给定字符串中，交换AB的功能
void f(char* data)
{
    int i;
    int tag;
    int dd = 0; // 移动方向

    while(1){
        tag = 0;
        //优先执行跳跃
        for(i=0; i<strlen(data); i++){
            if(data[i]=='.') continue;
            if(data[i]=='A') dd = 1;
            if(data[i]=='B') dd = -1;

            if(valid(data, i+dd) && valid(data,i+dd+dd)
            && data[i+dd]!=data[i] && data[i+dd+dd]=='.'){
            //如果能跳...
                move(data, i, i+dd+dd);
                printf("%s\n", data);
                tag = 1;
                break;
            }
        }

        if(tag) continue;
        //执行移动
        for(i=0; i<strlen(data); i++){
            if(data[i]=='.') continue;
            if(data[i]=='A') dd = 1;
            if(data[i]=='B') dd = -1;

            if(valid(data, i+dd) && data[i+dd]=='.'){
            // 如果能移动...
                if(valid(data, i-dd) && valid(data,i+dd*2) && data[i-dd] == data[i+dd*2]) continue;
                move(data, i, i+dd);
                printf("%s\n", data);
                tag = 1;
                break;
            }
        }

        if(tag==0) break;
    }
}

int main()
{
    char data[] = "AAA.BBB";
    f(data);
    return 0;
}