牛客---坐标移动

开发一个坐标计算工具, 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)

import sys

for line in sys.stdin:
    a = line.strip().split(';')
    b = a[:-1]
    v = [0,0]
    c = []
    for i in b:
        if len(i)>1 and len(i)<=3:
            if i[0] in 'ADWS' and i[1:].isdigit() == True:
                c.append(i)
    for j in c:
        if j[0] == 'A':
            v[0] = v[0]-int(j[1:])
        elif j[0] == 'D':
            v[0] = v[0]+int(j[1:])
        elif j[0] == 'W':
            v[1] = v[1]+int(j[1:])
        else:
            v[1] = v[1]-int(j[1:])
    print(','.join([str(i) for i in v]))

你可能感兴趣的:(python)