ฉันจะย่อรหัสหลามนี้ได้อย่างไร

นี่คือรหัสที่ฉันต้องการย่อ

n=input()
while n:
 s=raw_input()
 x,r,g,b=(int(x) for x in s.split())
 a=x/r%2
 c=x/g%2
 d=x/b%2
 r=((a*10+c)*10)+d
 if r==0:e="black"
 elif r==100:e="red"
 elif r==1:e="blue"
 elif r==10:e="green"
 elif r==101:e="magenta"
 elif r==11:e="cyan"
 elif r==110:e="yellow"
 else:e="white"
 print(e)
 n-=1

อินพุต: 3

4643 5913 4827 9752
5583 5357 5120 9400
2025 5475 4339 8392

เอาท์พุท:

black
yellow
black

แทนที่จะใช้((a*10+c)*10)+dเราสามารถ((a*2+c)*2)+dแยกแยะสีได้

 r=((a*2+c)*2)+d
 if r==0:e="black"
 elif r==4:e="red"
 elif r==1:e="blue"
 elif r==2:e="green"
 elif r==5:e="magenta"
 elif r==3:e="cyan"
 elif r==6:e="yellow"
 else:e="white"

อา แต่ตอนนี้เราแค่แยกความแตกต่างระหว่างค่าจาก0ถึง7ดังนั้นเราจึงสามารถจัดทำดัชนีเป็นอาร์เรย์แทนได้!

 r=a*4+c*2+d
 e=["black","blue","green","cyan","red","magenta","yellow","white"][r]
 # or even shorter:
 e="black blue green cyan red magenta yellow white".split()[r]

รวมกับการเปลี่ยนแปลงยูเรียลของเราได้รับลงไป 136 ไบต์(164 ไบต์บันทึกไว้)

exec'x,r,g,b=map(int,raw_input().split());print"black blue green cyan red magenta yellow white".split()[x/r%2*4+x/g%2*2+x/b%2];'*input()

ลองออนไลน์!

สำหรับการทำซ้ำใช้execคำสั่ง

map(int, สำหรับการแปลงอินพุตสตริงเป็นตัวเลข

ร่นคำนวณrด้วยr=a*100+c*10+dแล้วใส่คำนวณของแต่ละตัวแปร ( a, c, d) แทนตัวแปร

และสำหรับเงื่อนไขให้ใช้พจนานุกรมที่มีgetคิวรี

ในที่สุดก็บดทุกอย่างให้เป็นหนึ่งบรรทัด

ผลลัพธ์สุดท้าย (อัปเดต):

exec'x,r,g,b=map(int,raw_input().split());print({0:"black",100:"red",1:"blue",10:"green",101:"magenta",11:"cyan",110:"yellow"}.get((x/r%2)*100+(x/g%2)*10+x/b%2,"white"));'*input()

ไบต์บันทึก: 121