1027.プリント砂時計(20)--Python

問題の構想はとても簡単で、開始時にテストポイント1が合格しなかったのは、残りが0の時に戻った値がうまくいかなかったためで、前に別の方法を使ったので、変更を忘れました.最初はなぜ入力0をテストしなかったのかと思っていたが、后で问题を见ると、正の整数を入力するように要求され、めまいがした...
今まで、明らかにこれらの小さなプログラムを書くのに手応えがあったと感じて、続けて、私たちとして、コード行が重要です.

#          

def get_max_num(num, remainder):
	
	sum = 1 
	add = 1
	while sum < num:
		add += 2
		sum += add * 2
	
	if sum != num:
		sum = sum - add * 2
		add = add - 2
	remainder = num - sum
	out_flag(flag, add)
	
	print(remainder)
	
def out_flag(flag, flag_1):
	flag_2 = flag_1
	flag_3 = 0

	while True:
		if flag_2 == -1:
			break
		print(" " * flag_3 + flag * flag_2)
		flag_3 += 1
		flag_2 -= 2
	
	flag_2 += 4
	flag_3 -= 2
	while True:
		if flag_2 == flag_1 + 2:
			break
		print(" " * flag_3 + flag * flag_2)
		flag_3 -= 1
		flag_2 += 2
num, flag = input().split()
num = int(num)

remainder = 0
get_max_num(num, remainder)