【python】正则表达式秒杀：牛客网-HJ17 坐标移动

描述

开发一个坐标计算工具， A表示向左移动，D表示向右移动，W表示向上移动，S表示向下移动。从（0,0）点开始移动，从输入字符串里面读取一些坐标，并将最终输入结果输出到输出文件里面。

输入：
合法坐标为A(或者D或者W或者S) + 数字（两位以内）
坐标之间以;分隔。
非法坐标点需要进行丢弃。如AA10; A1A; $%$; YAD; 等。
下面是一个简单的例子 如：

A10;S20;W10;D30;X;A1A;B10A11;;A10;
处理过程：
起点（0,0）

• A10 = （-10,0）

• S20 = (-10,-20)

• W10 = (-10,-10)

• D30 = (20,-10)

• x = 无效

• A1A = 无效

• B10A11 = 无效

• 一个空 不影响

• A10 = (10,-10)
  结果 （10， -10）

数据范围：每组输入的字符串长度满足 1≤n≤10000 1≤n≤10000 ，坐标保证满足 −231≤x,y≤231−1 −231≤x,y≤231−1 ，且数字部分仅含正数

输入描述：
一行字符串
输出描述：
最终坐标，以逗号分隔

示例1
输入：
A10;S20;W10;D30;X;A1A;B10A11;;A10;
输出：
10,-10

示例2
输入：
ABC;AKL;DA1;
输出：
0,0

代码

import re
xy_coord = [0,0]
vector_map = {'W':(0,1),'A':(-1,0),'S':(0,-1),'D':(1,0)}
for input_word in input().split(";"):
    fd = re.findall('^(\w)(\d+)$',input_word)
    if(len(fd) == 0 or fd[0][0] not in vector_map): continue
    for i in range(2): xy_coord[i] += vector_map[fd[0][0]][i]*int(fd[0][1])
print(str(xy_coord).replace("[","").replace("]","").replace(" ",""))