初め、aの存在範囲を勝手に-50 ~ 50としていて総当たりしたが間違いだと気が付く。 あくまでもa + bが0 <= a + b <= 100なのであってこれからa, bの存在範囲は決められない。
では違うアプローチ。 表のすべての値を足すとsum(a1, a2, a3, b1, b2, b3) * 3、というのはすぐわかる。3の倍数が出てきてうれしくなるが、これだけでは必要条件とはならなそう。 sum(a1, a2, a3, b1, b2, b3)がどこかにあるかと探すと、c11, c22, c33があった。 これらが3倍になっていることを確認すると、9つの数値が使われているのでよさそう、で実際よかった。
証明まで持ち込めはしないが直観的にはこれでいいか。
def solve(c1, c2, c3): if (c1[0] + c2[1] + c3[2]) * 3 == sum(c1) + sum(c2) + sum(c3): return "Yes" return "No" if __name__ == "__main__": c1 = list(map(int, input().split(" "))) c2 = list(map(int, input().split(" "))) c3 = list(map(int, input().split(" "))) print(solve(c1, c2, c3))
