蓝桥杯 平方和拆分

cnt = 0
for i in range(1,2022):
    flag = 0
    for j in range(0,50):
        for k in range(0,50):
            if i == j*j + k*k:
                cnt += 1
                flag = 1
                break
        if flag == 1:
            break
    
print(cnt)

感悟

k循环的时候不能以i+1开始，因为可能存在i = j*j + j*j的情况存在