python どう記述したらよいのか(AtCoder Beginner Contest188-C)

Question

At coderのABC188のC問題(https://atcoder.jp/contests/abc188/tasks/abc188_c)
を解いたのですがACできません。

n=int(input())
c=list(map(int,input().split()))
c=a
b=[]
while len(a)>2:
  for i in range(0,len(a)):
    if i%2!=0:
      continue
    else:
      if a[i]>a[i+1]:
        b.append(a[i+1])
      else:
        b.append(a[i])
  for j in range(len(b)):
    a.remove(b[j])
  b.clear()
d=sorted(a)
for m in range(len(c)):
  if d[0]==c[m]:
    print(m+1)
    break

どこを直したらAC出来るのでしょうか。または、https://qiita.com/u2dayo/items/217dad4c9ba3fddd349d
解答のように書かないと出来ないのでしょうか。

Answer 1

いきなりaという未定義の変数が出てきますが、これはおそらく

-c=list(map(int,input().split()))
+a=list(map(int,input().split()))

の間違いですね。

その上で、このコードの問題点は、その直後のc=aにあります。これは、処理の過程で要素が削除されるaの初期状態を保存することを意図しているのだと思いますが、実際にはaとcは同一のオブジェクトを参照しているため、aのある要素が削除されると、cの同じ要素も削除されます。

aの初期状態を保存しておきたいのであれば、aをコピーしてcに設定してください。

-c=a
+c=a.copy()

これで問題の入力例に対しては正しい答えが得られます。実際の問題に対してはタイムリミット超えになるものがでるのでは。

Comments

[unknock]

ありがとうございます。動きました。
(ちなみにpythonでは時間超過でしたがpypyではセーフでした)

[Daregada]

おそらく、

n = int(input())
lst = list(map(int, input().split()))
x = len(lst) // 2
print(lst.index(min(max(lst[:x]), max(lst[x:]))) + 1)

みたいなのが、時間内に答えを得るためのコードかな。

[unknock]

あ、返信気づきませんでした。
なるほど、、こんなに短く回答できたのですね(+_+)
というか真ん中で切ってそれぞれで最大のもの同士戦わせればよかったのか。
丁寧にありがとうございます。

Answer 2

c=a

のところで、aが未定義というエラーが出ます。
ご自身の実行環境では、この前にaを定義しているのではないでしょうか。

a=c

に変更すれば動くかもしれませんね。