昨日ABCでなぜかE問題がさくっと解けたのでいろいろ試してたらなんと1行で書けてしまったのでメモとして。
該当の問題
def main():
T = int(input())
for _ in range(T):
A = input()
B = input()
a = A+A
try:
print(a.index(B))
except ValueError:
print(-1)
main()
print("\n".join([str((input()*2).find(input())) for _ in range(int(input()))]))
解説
問題はこんな感じ。
0, 1 からなる長さの等しい文字列 A,B が与えられます。
A に対して以下の操作を 0 回以上何度でも行うことができます。
- A の先頭の文字を末尾に移動させる
A=B とするために必要な最小の操作回数を求めてください。
但し、どのように操作しても A=B とできない場合、代わりに −1 と出力してください。T 個のテストケースが与えられるので、それぞれについて答えを求めてください。
例えば A=test,B=ttes だとこの操作が3回必要となる。 (test->estt->stte->ttes)
この操作ってtesttest (Aを2回繰り返した文字列)のなかから順にみてけばいける感じである。(語彙力不足)
"test "
" estt " #見る部分をずらしていく感じ
" stte "
" ttes " #ここで一致する!
" test"
なのでtesttestからttesを探す問題へと変化する。
Pythonにはstr.find() という素晴らしい関数が存在するのでそれを使ってあげれば問題ない。
しかも見つからなかった時に-1を返すという部分まで実装されているのは素晴らしい。
ここまでくればコンテスト中にだしたコードは理解できるでしょう。
def main():
T = int(input()) #テストケース数を受け取る
for _ in range(T): #テストケースの数だけ繰り返す
A = input() #Aの入力
B = input() #Bの入力
a = A+A #Aを2回繰り返す
try:
print(a.index(B)) # Bの場所を探して出力
except ValueError: #str.index()は見つからなかったときに例外を送出するのでそれをキャッチ
print(-1) #-1を出力
main()
ここからは改善していくだけ。
ここで
try:
print(a.index(B))
except ValueError:
print(-1)
の部分は print(a.find(B)) に置き換えることができる。
def main():
T = int(input())
for _ in range(T):
A = input()
B = input()
a = A+A
print(a.find(B)) #findに置き換えた
main()
わざわざmain()に関数化する必要はないので削除。
T = int(input())
for _ in range(T):
A = input()
B = input()
a = A+A
print(a.find(B))
A+AはA*2と書けるので
T = int(input())
for _ in range(T):
A = input()
B = input()
a = A*2 #文字列*数字 は文字列を数字回繰り返したものになる ("abc"*3 = "abcabcabc")
print(a.find(B))
T,B Aは1回しかつかわないので使うときに取得
for _ in range(int(input())):
a = input()*2
print(a.find(input()))
a.find(input())のfind()内のinput()よりもaが先に読み込まれるため、aも詰められる
for _ in range(int(input())):
print((input()*2).find(input()))
最後に一括で出力するように変更してみる。
str.join()に入れる引数はlist[int]だとエラーがでるのでlist[str]を入れる。
r = []
for _ in range(int(input())):
r.append(str((input()*2).find(input())))
print("\n".join(a)) #改行で区切られている
ここでrについて考えてみると、リスト内表記が使えることに気付く。
#hogehogeに何らかの処理、Nに数字を入れる
r = []
for i in range(N)
r.append(hogehoge)
# 上と下は全く同じ結果になる(はず)
r = [hogehoge for i in range(N)]
だから
r = [str((input()*2).find(input())) for _ in range(int(input()))]
print("\n".join(r))
rもまとめられるので
print("\n".join([str((input()*2).find(input())) for _ in range(int(input()))]))
おお。なんと1行で書けてしまったのである。
コンテスト中にE問題を1行で提出したらロマンありそうだなぁと思った。特にPythonは改行とインデントに意味があるので難しいだろう。
これ思いつけそうだったのが悔しい。index()じゃなくてfind()を先に見つけていれば書けそうだったなぁ
終わりに
運営もしかして狙ったか?