[Python知识库] 【kickstart 2021 roundF】第一题 trash bins Python题解

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> Python知识库 -> 【kickstart 2021 roundF】第一题 trash bins Python题解 -> 正文阅读

[Python知识库]【kickstart 2021 roundF】第一题 trash bins Python题解

from collections import defaultdict
T=int(input())
for tt in range(T):
    N=int(input())
    ss=input()
    ans=0
    
    ll=[-999999 for nn in range(N)]
    rr=[9999999 for nn in range(N)]
    if ss[0]=='1':
        ll[0]=0
    for nn in range(1,N):
        if ss[nn]=='0':
            ll[nn]=ll[nn-1]
        else:
            ll[nn]=nn
    if ss[N-1]=='1':
        rr[N-1]=N-1
    for nn in range(N-2,-1,-1):
        if ss[nn]=='0':
            rr[nn]=rr[nn+1]
        else:
            rr[nn]=nn
    for nn in range(N):
        ans+=min(nn-ll[nn],rr[nn]-nn)
    print('Case #%d: %d' % (tt+1, ans))