【华为OD题库-054】最小循环子数组-java

题目

给定一个由若干整数组成的数组nums，请检查数组是否是由某个子数组重复循环拼接而成，请输出这个最小的子数组。
输入描述
第一行输入数组中元素个数n，1 <= n <= 100000
第二行输入数组的数字序列nums，以空格分割，0 <= nums[i]<= 10
输出描述
输出最小的子数组的数字序列，以空格分割;
备注
数组本身是其最大的子数组，循环1次可生成的自身
示例1:
输入
9
1 2 1 1 2 1 1 2 1
输出
1 2 1
说明
数组[1,2,1,1,2,1,1,2,1]可由子数组[1,2,1]重复循环3次拼接而成

思路

暴力查找

依次遍历nums，检查nums是否能够有0~i的子串循环拼接而成（checked方法）
checked逻辑，假定nums可以由tmp循环拼接而成，那么遍历nums，对于每一位i，其值应该等于tmp中对应位置的值，即：nums[i]=tmp[i%length]，length为tmp的长度。
还需要注意，如果nums可以由tmp循环拼接而成，那么nums的长度一定是tmp长度的整数倍。

转字符串匹配

类似leetcode: 459. 重复的子字符串

因为输入的是0~10的数字，可以将他们转成字符串，比如：原来的输入是：1 10 2 1 10 2，记为s1
两个s1加空格拼接后的字符串s2: 1 10 2 1 10 2 1 10 2 1 10 2
从s2中查找第二次出现s1的索引位置，即：s2.indexOf(s1,1)，1 10 2 1 10 2 1 10 2 1 10 2，标黄色部分为第二次出现s1的子串，所以循环字符就是其前面的字符串：1 10 2，直接输出即可。

题解

暴力查找

package hwod;
import java.util.Arrays;
import java.util.Scanner;

public class MinLoopArr {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = Integer.parseInt(sc.nextLine());
        int[] nums = new int[n];
        for (int i = 0; i < n; i++) {
            nums[i] = sc.nextInt();
        }
        int[] res = minLoopArr(nums);
        for (int i = 0; i < res.length; i++) {
            if (i != 0) System.out.print(" ");
            System.out.print(res[i]);
        }
    }

    private static int[] minLoopArr(int[] nums) {
        for (int i = 0; i < nums.length/2; i++) {
            if (checked(nums, i)) {
                return Arrays.copyOfRange(nums, 0, i + 1);
            }
        }
        return nums;
    }

    private static boolean checked(int[] nums, int i) {
        int[] tmp = Arrays.copyOfRange(nums, 0, i + 1);
        if (nums.length % tmp.length != 0) return false;
        for (int j = 0; j < nums.length; j++) {
            if (nums[j] != tmp[j % tmp.length]) return false;
        }
        return true;
    }
}

转字符串匹配

package hwod;
import java.util.Arrays;
import java.util.Scanner;

public class MinLoopArr {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = Integer.parseInt(sc.nextLine());
        String originStr = sc.nextLine();
        System.out.println(minLoopArr(originStr));
    }

    private static String minLoopArr(String s) {
        int idx = (s +" "+ s).indexOf(s, 1);
        idx = Math.min(s.length(), idx);
        return s.substring(0, idx).trim();
    }
}

注意：leetcode官方题解还给出了KMP算法的思路，暂时没看懂，后面再学习

推荐

如果你对本系列的其他题目感兴趣，可以参考华为OD机试真题及题解（JAVA），查看当前专栏更新的所有题目。