4.2 进阶题 - B 相生相克

古代先民认为，天下万物皆由五类元素组成，分别是金、木、水、火、土，彼此之间存在相生相克的关系。

相生关系为：木生火，火生土，土生金，金生水，水生木。

相克关系为：金克木，木克土，土克水，水克火，火克金。

本题就请你编写程序，判断任意一对给定的元素之间的关系。

输入格式：

输入在一行中给出一个正整数 N（≤10），随后 N 行，每行给出
2 个正整数 A 和 B，为两种元素的编号。这里假设金、木、水、火、土的编号顺次为 1、2、3、4、5。

输出格式：

对输入的每一对 A 和 B，如果 A 生 B，则输出 A sheng B；如果 A 克 B，则输出 A ke B。反之亦然。

输入样例：

2
4 2
5 3

输出样例：

2 sheng 4
5 ke 3

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

def shengke(n):
   sheng={"1":3,
          "2":4,
          "3":2,
          "4":5,
          "5":1,}
   ke={"1":2,
       "2":5,
       "3":4,
       "4":1,
       "5":3,

   }
   p=[]
   for i in range(n):
       p.append(input())

   for i in p:
       a,b=i.split(" ")
       if int(b)==sheng[a]:
           print(a+" sheng "+b)
       if int(a)==sheng[b]:
            print(b + " sheng " + a)
       if int(b)==ke[a]:
           print(a+" ke "+b)
       if int(a)==ke[b]:
            print(b + " ke " + a)
shengke(int(input()))