PTA(BasicLevel)-1031 查验身份证

一、问题定义

一个合法的身份证号码由17位地区、日期编号和顺序编号加1位校验码组成。校验码的计算规则如下：
首先对前17位数字加权求和，权重分配为：{7，9，10，5，8，4，2，1，6，3，7，9，10，5，8，4，2}；
然后将计算的和对11取模得到值Z；最后按照以下关系对应Z值与校验码M的值：

Z：0 1 2 3 4 5 6 7 8 9 10
M：1 0 X 9 8 7 6 5 4 3 2

现在给定一些身份证号码，请你验证校验码的有效性，并输出有问题的号码。

输入格式：
输入第一行给出正整数N（≤100）是输入的身份证号码的个数。随后N行，每行给出1个18位身份证号码。

输出格式：
按照输入的顺序每行输出1个有问题的身份证号码。
这里并不检验前17位是否合理，只检查前17位是否全为数字且最后1位校验码计算准确。如果所有号码都正常，则输出All passed。

输入样例 1：

4
320124198808240056
12010X198901011234
110108196711301866
37070419881216001X

输出样例 1：

12010X198901011234
110108196711301866
37070419881216001X

输入样例 2：

2
320124198808240056
110108196711301862

输出样例 2：

All passed

 

二、解题

# 1031 check identity number
num = int(input())
id_nums = []
for i in range(num):
    id_nums.append(input())

W = [ 7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2]
M = [i for i in "1 0 X 9 8 7 6 5 4 3 2".split(' ')]

def get_code(id):
    mod_num = 11
    pre_num = [i for i in id[:-1]]
    for j in [chr(i + ord('A')) for i in range(26)]:
        if j in pre_num:
            return -1
    else:
        pre_num = [int(p) for p in pre_num]
        sum = 0
        for i in range(len(W)):
            sum += pre_num[i] * W[i]
        
        return sum % mod_num
        
        
flag = len(id_nums)
for id in id_nums:
    check_num = get_code(id) 
    # 17位中有一个'X'
    if check_num == -1:
        flag -= 1
        print(id)
    # check_num in 0 to 10, exam check_num
    elif M[check_num] != id[-1]:
        # id is a string that contians alpha and number, must compare char to char
        flag -= 1
        print(id)

if flag == len(id_nums):
    print("All passed")