题目
给定一个由若干整数组成的数组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,直接输出即可。
题解
暴力查找
java
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;
}
}
转字符串匹配
java
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),查看当前专栏更新的所有题目。