51nod 1554 欧姆诺姆和项链（KMP+思路）

最新推荐文章于 2021-06-01 13:19:11 发布

Gijkstra

最新推荐文章于 2021-06-01 13:19:11 发布

阅读量298

点赞数

分类专栏：【KMP&&扩展】

本文链接：https://blog.csdn.net/m0_37134257/article/details/79831140

版权

【KMP&&扩展】专栏收录该内容

8 篇文章 0 订阅

订阅专栏

1554 欧姆诺姆和项链

题目来源： CodeForces

基准时间限制：1 秒空间限制：131072 KB 分值: 80 难度：5级算法题

有一天，欧姆诺姆发现了一串长度为n的宝石串，上面有五颜六色的宝石。他决定摘取前面若干个宝石来做成一个漂亮的项链。

他对漂亮的项链是这样定义的，现在有一条项链S，当S=A+B+A+B+A+...+A+B+A的时候是漂亮的，这儿A，B是一些宝石串，“+”表示连接操作。S中有k+1个A和k个B组成。A和B可能是空串。

现在给出宝石串，问怎么切前几个才能得到一个漂亮的宝石项链。他切下来之后不会改变宝石的顺序。

样例解释：

在这个样例中前6个可以组成漂亮的串（ A="", B="bca"）。前7个也可以（A="b", B="ca"）。

Input

单组测试数据。
第一行有两个整数n, k (1≤n,k≤1 000 000)，表示宝石串原始的长度和在上文中提到的参数k。
第二行有n个由小写字母组成的串，表示原始宝石串。

Output

输出一行有n个01组成的字符串。第i(1≤i≤n)个位置是1的时候表示前i个宝石可以组成漂亮的宝石项链。

Input示例

样例输入1
7 2
bcabcab

Output示例

样例输出1
0000011

中文题就不说题意了

首先利用kmp的next数组的性质，可以求出1~i的循环节长度

设i点为当前点，每个循环节为S，那么一共有两种情况

1：循环节可以完全把1~i覆盖，即SSSS...SS的情况

然后开始考虑在这么多循环节中选k+1个A，k个B

因为一定要选k个AB

计算出每个AB串中含有 i / ( i - next[i] ) / k 个s,最后会剩下 i / ( i - next[i] ) % k 个s，

剩下s的个数<=每个AB中含有的S个数,则可以被覆盖（将剩下的s看做A，此时就有了k+1个A，k个B）

如果剩下s的个数>每个AB中含有的S个数，此时i就不能被覆盖，因为还没到i，就选够了所需的k~k+1个

2：循环节不能完全把1~i覆盖，SSSS...ST的情况，T为S的一个前缀

此时只有k+1个A个k个B的情况，因为多了个额外的T串

那么A一定=T，其余过程都和上面一样，但注意一点，最后的剩下s的个数需<每个AB中含有的S个数

这时为什么不能=了？因为之前的最后没有T，=的情况，可以吧A看做剩下的S，而现在有了T，剩下的S+T则大于原本的循环节了

#include<stdio.h>
const int N=1e6+2005;
int nxt[N],n,m;
char s[N];
void getnxt()
{
    int j=-1,i=0;
    nxt[0]=-1;
    while(i<n)
    {
        if(j==-1||s[i]==s[j])
        nxt[++i]=++j;
        else
        j=nxt[j];
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    scanf("%s",s);
    getnxt();
    for(int i=1;i<=n;i++)
    {
        int x=i/(i-nxt[i]);//i的循环节出现次数，即S的个数
        if(i%(i-nxt[i]))//i不能被完全覆盖
        {
            //此时为SSS...T的情况，T为s串的一个前缀
            if(x/m>x%m)printf("1");
            else printf("0");
        }
        else//i被完全覆盖
        {
            //此时为SSS...S的情况
            //k=x/m,为每个AB所占的S个数，此时可设B为空串，选取m+1个A，故可以相等
            if(x/m>=x%m)printf("1");
            else printf("0");
        }
    }
    puts("");
    return 0;
}