C++ 将头朝向转化为数组 简便思路

解题思路:可以将蚂蚁左或右转向再向前移动转化为:蚂蚁移动到上下左右四个方向上.关键是        上            可以转化为数组dir={左,上,右,下};即dir={0,1,2,3}

                                                                                                                                         左        右

                                                                                                                                                下

再配合方向值

dx[4]={0,-1,0,1},dy[4]={-1,0,1,0};

即dir[ind]代表头部的朝向,将向左看为ind-1,向右看为ind+1,再通过公式:ind=(4+ind+1(向左就是-1))%4保证ind值在0-3内

则

            x+=dx[ind];

            y+=dy[ind];

就代表蚂蚁移动到的位置.
注意事项:

参考代码:

#include<iostream>
#include <vector>
using namespace std;
int main()
{
    int m,n,x,y,k,ind;
    int dx[4]={0,-1,0,1},dy[4]={-1,0,1,0};
    char direction;
    cin>>m>>n;
    vector<vector<int>> a(m,vector<int>(n));
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            cin>>a[i][j];
        }
    }
    cin>>x>>y>>direction>>k;
    ind = direction=='U'? 1:direction=='D'? 3:direction=='L'? 0:2;
    while(k--){
        if(a[x][y]){
            a[x][y]=0;
            ind=(4+ind+1)%4;
            x+=dx[ind];
            y+=dy[ind];
        }
        else{
            a[x][y]=1;
            ind=(4+ind-1)%4;
            x+=dx[ind];
            y+=dy[ind];
        }
    }
    cout<<x<<" "<<y;
    return 0;
}