P3392 涂国旗 python解法

P3392 涂国旗 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

n,m=map(int,input().split())
arr=[]
ks=[]
ans=100000000 #因为要比较最小值，这里由于数据较小可以设置这么大，数据要再大的话就设置最大常量
#前i行涂蓝成本b[i],涂红成本r[i],涂白成本w[i],求总成本最小值min(w[i]+b[j]-b[i]+r[n]-r[j])
#这里因为n<=50,所以只用开这么大就行了
w=[0 for _ in range(51)]
b=[0 for _ in range(51)]
r=[0 for _ in range(51)]
k=0
for i in range(1,n+1):
    ks=list(input())
    arr.append(ks)
    w[i]=w[i-1]+(m-ks.count('W'))
    b[i]=b[i-1]+(m-ks.count('B'))
    r[i]=r[i-1]+(m-ks.count('R'))
#i为白蓝分界线（i行还是白），j为蓝红分界线（j行还是蓝）,通过枚举分界线来判断最小成本
for i in range(1,n-1):
    for j in range(i+1,n):
        ans=min(ans,w[i]+(b[j]-b[i])+(r[n]-r[j])) 
print(ans)