[AtCoder] ABC 249 A-Jogging

AtCoder

Photo by Daniel Reche on Pexels.com

2022.05.07

Table of Contents

問題

https://atcoder.jp/contests/abc249/tasks/abc249_a

方針

計算で求めようとしたが（A問題としては）難しく、以下のようにリストを作成した。

p = [b]*a + [0]*c

このリストから最初のx秒を取って、和を計算すればよい。

解答

#atcoder template
def main():
    import sys
    input = sys.stdin.readline
    #文字列入力の時は上記はerrorとなる。
    #ここにコード
    #input
    a, b, c, d, e, f, x= map(int, input().split())

    #output
    p = [b]*a + [0]*c
    q = [e]*d + [0]*f
    P = p*100
    Q = q*100
    if sum(P[:x]) > sum(Q[:x]):
        print("Takahashi")
    elif sum(P[:x]) < sum(Q[:x]):
        print("Aoki")
    else:
        print("Draw")

    #N = 1のときなどcorner caseを確認！
if __name__ == "__main__":
    main()