PTA L1-062 幸运彩票

彩票的号码有 6 位数字，若一张彩票的前 3 位上的数之和等于后 3 位上的数之和，则称这张彩票是幸运的。本题就请你判断给定的彩票是不是幸运的。

输入格式：

输入在第一行中给出一个正整数 N（≤ 100）。随后 N 行，每行给出一张彩票的 6 位数字。

输出格式：

对每张彩票，如果它是幸运的，就在一行中输出 You are lucky!；否则输出 Wish you good luck.。

输入样例：

2
233008
123456

输出样例：

You are lucky!
Wish you good luck.

代码如下：

n = int(input())
for i in range(n):
    a = input()
    sum1 = 0
    sum2 = 0
    a1 = a[:3] #彩票的前3位数
    a2 = a[3:] #彩票的后3位数
    for i in a1:
        sum1 += int(i) #前3位数累加
    for j in a2:
        sum2 += int(j) #后3位数累加
    if sum1 == sum2:
        print("You are lucky!")
    else:
        print("Wish you good luck.")