需要优化此段代码import sysj=''primary,annex = {},{}n,m=0,0for line in sys.stdin:    a = line.split()    if len(a) == 2:        n,m=map(int,a)    if len(a) == 3:        x,y,z = map(int,a)        if z==0:            primary[len(j)] = [x,y]        else:            if z in annex:                annex[z].append(...