taketakeyyy / atcoder

my atcoder codes.

Bow there 🐶

🐾taketakeyyy.github.io

powered by github-profile-summary-cards.

atcoder's Issues

ABC140のD問題で、コーナーケースっぽいやつでコケる

解説PDFをシミュレートするコードが、コーナーケースっぽい testcase_19 でコケるので、テストケースが公開されたら再考する。

def solve():
    N, K = list(map(int, input().split()))
    S = input()
    S = "R" + S + "L"

    endRL = 0  # 端っこのRL
    RL, LR = 0, 0  # RLの数, LRの数
    happy_num = 0  # 幸福な人の数

    M = len(S)
    for i in range(1, M):
        if S[i-1]=="R" and S[i]=="L":
            if (i-1)==0 or i==(M-1):
                endRL += 1
            else:
                RL += 1
        elif S[i-1]=="L" and S[i]=="R":
            LR += 1

        # 初期状態の幸福な人の数を数える
        if (i-1)==0 or i==(M-1):
            continue
        if (S[i-1]=="L" and S[i]=="L") or (S[i-1]=="R" and S[i]=="R"):
            happy_num += 1

    # 幸福な人の数を可能な限り最大化する
    if RL >= LR:
        if K >= LR:
            happy_num += LR*2
            RL -= LR
            K -= LR
        else:
            happy_num += K*2
    else:
        if K >= RL:
            happy_num += RL*2
            LR -= RL
            K -= RL
            if K >= min(LR, endRL):
                happy_num += min(LR, endRL)
            else:
                happy_num += K
        else:
            happy_num += K*2

    print(happy_num)

ABC131のC問題、floor()だとWA、//演算子だとACなのなんで？

テストケースが開放されたら検証する