【AtCoderネタ】でぇじょうぶだ　オラにはまだPyPIがある

今回はAtCoderネタです。ターゲットは以下の通り、

んで、こんなコードを書いてみる

K, S = map(int, input().split())
count = 0
for x in range(K+1):
    for y in range(K+1):
        z = S - x - y
        if x + y + z == S and 0 <= z <= K:
            #print('{} {} {}'.format(x,y,z))
            count += 1

print(count)

この問題はO(n^2)のコードで公式解となっています。

このコードもO(n^2)で実装しています。

ぶっちゃけPythonでギリギリです。

んで実はギリギリどころか無理なんすよねぇ

それで提出するとさ、

こうなっちゃうわけです・・・

ヒャッハァ〜！！TLE解ダァ〜〜！！

んで正解のテストケースをよく見てくれ。

subtask_1_03.txt 1892 ms

こりゃあもうダメかもしれんね。

まだだ　まだ策がある

いやちょっと待って欲しい我らPythonistaには解法がまだあるじゃないかここで本当に諦めてええんか・・・？

あるやろ、PyPIが

よっしゃあじゃあぶちこむぜ！！！

reud, WIN!

完

まだだ まだ策がある

まだだ　まだ策がある