华为OD 最小循环子数组

发布于:2025-06-22 ⋅ 阅读:(14) ⋅ 点赞:(0)

1. 题意

给定一个由若干整数组成的数组 nums,请检查数组是否是由某个子数组重复循环拼接而成,请输出这个最小的子数组。

2. 题解

利用 k m p kmp kmp中的 n e x t next next数组性质,我们可以求出 n u m s nums nums中的最长公共

前缀后缀。如果要是由一个子数组循环组成的话,那么必然有 n e x t [ s z ] % ( n − n e x t [ s z ] ) = 0 next[sz] \% (n-next[sz]) =0 next[sz]%(nnext[sz])=0

为什么满足这个性质就能说明是由子数组循环若干次组成的呢?

我们假设上面的条件成立,且将字符串每 k = n − n e x t [ n ] k = n-next[n] k=nnext[n]个划为一组。那么字符串可以表示为 b 1 ⋯   b n / k b_1\cdots\ b_{n/k} b1 bn/k

由于 k = n − n e x t [ n ] k=n-next[n] k=nnext[n], 因此串 b 0 b 1 ⋯ b n / k − 1 = b 2 ⋯ b n / k b_0b_1\cdots b_{n/k-1}=b_2\cdots b_{n/k} b0b1bn/k1=b2bn/k(最长公共前后缀)。进而
b 1 = b 2 b 2 = b 3 ⋯ b n / k − 1 = b n / k b_1=b_2\\ b_2=b_3\\ \cdots\\ b_{n/k-1}=b_{n/k} b1=b2b2=b3bn/k1=bn/k
因此
b 1 = b 2 = ⋯ = b n / k b_1=b_2=\cdots=b_{n/k} b1=b2==bn/k
串为循环串。

例如

s     :    a  b c a b c a b c
next  :    -1 0 0 0 1 2 3 4 5 6
n = 9
next[n] = 6
n -next[n] = 3
6 = 2 x 3
  • 代码
#include <iostream>
#include <string>
#include <vector>

int main()
{
    int n;
    std::cin >> n;

    std::vector<int> a(n, 0);

    for ( int i = 0; i < n; ++i) {
        std::cin >> a[i];
    }

    std::vector<int> next( n + 1, 0);
    int j = -1;
    int i =  0;
    next[0] = -1;

    while ( i < n) {
        if ( j == -1 || a[i] == a[j]) {
            ++i;
            ++j;
            next[i] = j;
        }
        else {
            j = next[j];
        }
    }

    int k = n;
    if ( (next[n] >= (n - next[n])) && (next[n] % (n - next[n]) == 0)) {
        k = n - next[n];
    }

    for (int i = 0; i < k; ++i) {
        if ( i )
            std::cout << " ";
        std::cout << a[i];
    }


    return 0;
}

网站公告

今日签到

点亮在社区的每一天
去签到