ABC009 C問題のWA要因について

Question

前提・実現したいこと

AtCoder Beginner Contest 009のC問題が通らず、WAとなっている要因を知りたいです。

[AtCoder Beginner Contest 009 C - 辞書式順序ふたたび]
https://atcoder.jp/contests/abc009/tasks/abc009_3

発生している問題・エラーメッセージ

テストケースのうち11個のみACとなっております。
問題文における入力例は全て通っており、NGとなるケースも特定出来ていない状況です。

該当のソースコード

言語：Python3 (3.4.3)

import copy,sys
input=sys.stdin.readline
N,K=[int(x) for x in input().split()]
S=list(input())
ans=[]
cost=0
remaining=copy.deepcopy(S)
remaining.sort()

def check(fix,w,S):
  res=0
  s=copy.deepcopy(S)

  for i in range(len(fix)):
    if fix[i]!=s[i]:
      res+=1
  s=s[len(fix):]
  for wd in w:
    if wd in s:
      s.remove(wd)
    else:
      res+=1
  return res

for i in range(N):
  for s in remaining:
    t_sortedS=copy.deepcopy(remaining)
    t_sortedS.remove(s)
    c=check(ans+[s],t_sortedS,S)
    if c>K:
      continue
    elif c<=K:
      ans.append(s)
      remaining.remove(s)
      break

print(''.join(ans))

(修正)

import copy,sys
input=sys.stdin.readline
N,K=[int(x) for x in input().split()]
S=list(input())
ans=[]
cost=0
remaining=copy.deepcopy(S)
remaining.sort()

def check(fix,w,S):
  res=0
  s=copy.deepcopy(S)

  for i in range(len(fix)):
    if fix[i]!=s[i]:
      res+=1
  s=s[len(fix):]
  seq=copy.deepcopy(w)
  for wd in seq:
    if wd in s:
      s.remove(wd)
    else:
      res+=1
  return res

for i in range(N):
  k=''
  seq=copy.deepcopy(remaining)
  for ii,s in enumerate(seq):
    t_sortedS=copy.deepcopy(remaining)
    t_sortedS.remove(s)
    c=check(ans+[s],t_sortedS,S)
    if c>K:
      continue
    elif c<=K:
      ans.append(s)
      k=s
      break

  remaining.remove(k)

print(''.join(ans))

試したこと

問題文中の入力例の他、自身で作成した入力サンプルにて出力結果が問題無いことを確認。

補足情報（FW/ツールのバージョンなど）

プログラミング初心者故、大変見にくいコードかと思います。
質問内容と関係の無い観点でも、改善点などありましたらアドバイスを頂けると幸いです。

Comments

[hayataka2049]

とりあえずインデントが消えて質問のコードが読めないので、編集画面を開いて<code>ボタンで挿入できるコードブロックの中に入れてください。

[H.K.]

早速のご確認ありがとうございます。該当箇所について修正いたしました。

Answer 1

pythonの仕様上、forとループ対象の変更を組み合わせると「変なこと」になります。

注釈 ループ中でのシーケンスの変更には微妙な問題があります (これはミュータブルなシーケンス、すなわちリストなどでのみ起こります)。どの要素が次に使われるかを追跡するために、内部的なカウンタが使われており、このカウンタは反復のたびに加算されます。このカウンタがシーケンスの長さに達すると、ループは終了します。このことから、スイート中でシーケンスから現在の (または以前の) 要素を除去すると、(次の要素のインデクスは、すでに取り扱った現在の要素のインデクスになるために) 次の要素が飛ばされることになります。（※筆者強調）　同様に、スイート中でシーケンス中の現在の要素以前に要素を挿入すると、現在の要素がループの次の週で再度扱われることになります。こうした仕様は、厄介なバグにつながります。

https://docs.python.org/ja/3/reference/compound_stmts.html#the-for-statement

　

参考（拙記事）：【python】listをforループで回してremoveしたら思い通りにならない - 静かなる名辞

その絡みで問題がないかよく確認してみるといいでしょう。

Comments

[H.K.]

回答を頂いた内容について確認しました。色々と修正を行ってみましたが、まだWAとなってしまいます。
対処として、for文のループ対象とするリストを事前に'copy.deepcopy()'にて複製を行っております。これにより、for文内での元々のリストへの変更は、forループの動作にも影響しない理解ですが、誤っておりますでしょうか。恐れ入りますが、可能であれば再度確認頂けると幸いです。
本文中(修正)に修正したコードを更新致しました。