Codeforces 1353E - K-periodic Garland

大概是我很懒的缘故,所以很喜欢看懒懒的猫咪,也许上辈子我是只大懒猫吧。

翻译

K周期花环

给你一个有n个灯的花环,灯的状态用长度为 n 的字符串 s 表示。如果第 i 个灯是开着的,那么 s[i] 就是 '0',如果第 i 个灯是关着的,那么 s[i] 就是 '1'。当然,也给你一个正整数 k。

每一次移动,你可以选择一个灯,改变它的状态。

如果这个花环每个相邻的开着的灯的距离都是 k,那么他被称为 k 周期花环。注意,这个花环不是一个圈,第一个开着的灯不在最后一个开着的灯后面。

你的任务是找到最少的移动步数,将花环变成 k 周期花环。

输入格式

第一行输入整数 t,表示有 t 组测试用例。

每个测试用例输入两行。

第一行输入整数 n k,用逗号分隔。

第二行输入长度为 n 的字符串,只包含 '0' 和 '1'。

可以保证,所有 n 的和不超过 10^6。

输出格式

输入最小的步数。

分析

这道题是一条动态规划题目,分析起来挺麻烦的,为此还写了很多debug代码。

动态规划需要做两次,分别是从左往右和从右往左,为了复用逻辑,第二遍就是翻转字符串再做一次从左往右的递归。

递推的结果就是,当前位置如果是 0 还有 1 时,分别需要多少步。假设当前位置需要改成 0,那么就为前 k 位也需要是 0 的情况,加上当前需要修改的次数。假设当前位置需要改成 1,那么前 k 位需要是 0 或者是 1 两者的最小值,因为不是循环,前导 0 可以符合条件。这也就是核心的递推公式。

以上需要计算每个位置的前 k - 1 位置有多少个 1,为了加速,会用滚动计算的方式预处理好前 k - 1 位包含 1 的数量。

给出一个很特殊的测试用例。

2
1 1
0
18 2
100000010100000001

答案分别是

0
2

第一个很好理解,第二个就是两边1都为 0 的情况。

代码(PyPy3)

# https://codeforces.com/contest/1353/problem/E

import sys
import os
import heapq

try:
    path = "./file/input.txt"
    if os.path.exists(path):
        sys.stdin = open(path, 'r')
    # sys.stdout = open(r"./file/output.txt", 'w')
except:
    pass

t = int(input())

def printd(value):
    # print(value)
    pass

for _ in range(t):
    arr = list(map(int, input().split(" ")))
    n, k = arr[0], arr[1]
    str = input()
    count = len(str)
    leftone = [0] * count
    leftzero = [0] * count
    rightone = [0] * count
    rightzero = [0] * count

    one = [0] * count
    # zero = [0] * count

    sum = 0
    for i, c in enumerate(str):
        if i > 0:
            sum += 1 if str[i - 1] == '1' else 0

        if i >= k:
            sum -= 1 if str[i - k] == '1' else 0

        one[i] = sum

    printd(str)
    printd(one)

    for i, c in enumerate(str):
        number = 1 if c == '0' else 0
        numberzero = 0 if c == '0' else 1
        if i >= k:
            number += min(leftone[i - k], leftzero[i - k])
            numberzero += leftzero[i - k]
        leftone[i] = one[i] + number
        leftzero[i] = one[i] + numberzero
    
    printd(leftone)
    printd(leftzero)

    printd("--right---")

    str = str[::-1]
    sum = 0
    for i, c in enumerate(str):
        if i > 0:
            sum += 1 if str[i - 1] == '1' else 0

        if i >= k:
            sum -= 1 if str[i - k] == '1' else 0

        one[i] = sum


    printd(str)
    printd(one)

    for i, c in enumerate(str):
        number = 1 if c == '0' else 0
        numberzero = 0 if c == '0' else 1
        if i >= k:
            number += min(rightone[i - k], rightzero[i - k])
            numberzero += rightzero[i - k]
        rightone[i] = one[i] + number
        rightzero[i] = one[i] + numberzero
    
    printd(rightone)
    printd(rightzero)

    leftone.reverse()
    leftzero.reverse()

    result = 99999999
    for i, c in enumerate(str):
        value = leftone[i] + rightone[i]
        value -= 1 if c == '0' else 0
        result = min(result, value)

        value = leftzero[i] + rightzero[i]
        value -= 0 if c == '0' else 1
        result = min(result, value)

    print(result)

更多代码尽在 https://github.com/Tconan99/Codeforces

by 费城的二鹏 2020.05.22 长春

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,919评论 6 502
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,567评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,316评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,294评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,318评论 6 390
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,245评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,120评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,964评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,376评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,592评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,764评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,460评论 5 344
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,070评论 3 327
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,697评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,846评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,819评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,665评论 2 354