题解:[蓝桥杯 2024 国 Python A/Java 研究生] 特别的数组
源题目地址:https://www.lanqiao.cn/problems/20149/learning/
题目分析
这道题目要求我们通过移除数组中的一个连续子数组,使得剩下的数组中的所有元素都互不相同。我们需要找到移除后能得到的最长特别数组。
解题思路
- 统计元素频率:首先统计数组中每个元素的出现频率
- 识别重复元素:找出所有出现超过一次的元素
- 滑动窗口:使用滑动窗口技术找到包含所有重复元素的最短连续子数组
- 计算最大长度:原数组长度减去最短移除窗口的长度即为答案
代码实现
import java.util.*;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
int n = scanner.nextInt();
int[] a = new int[n];
for (int i = 0; i < n; i++) {
a[i] = scanner.nextInt();
}
// 统计每个元素的总出现次数
Map<Integer, Integer> total = new HashMap<>();
for (int num : a) {
total.put(num, total.getOrDefault(num, 0) + 1);
}
// 收集所有重复元素(出现超过1次的元素)
Set<Integer> duplicates = new HashSet<>();
for (Map.Entry<Integer, Integer> entry : total.entrySet()) {
if (entry.getValue() > 1) {
duplicates.add(entry.getKey());
}
}
// 如果没有重复元素,直接返回原数组长度
if (duplicates.isEmpty()) {
System.out.println(n);
return;
}
// 滑动窗口寻找包含所有重复元素的最短子数组
Map<Integer, Integer> windowCount = new HashMap<>();
int left = 0, minWindow = Integer.MAX_VALUE;
int satisfied = 0; // 记录窗口中已经满足条件的重复元素数量
for (int right = 0; right < n; right++) {
int num = a[right];
if (duplicates.contains(num)) {
windowCount.put(num, windowCount.getOrDefault(num, 0) + 1);
// 当窗口中该元素的数量等于总出现次数减1时,表示可以移除这个元素的所有重复
if (windowCount.get(num) == total.get(num) - 1) {
satisfied++;
}
}
// 当窗口中包含所有重复元素的足够数量时,尝试缩小窗口
while (satisfied == duplicates.size()) {
minWindow = Math.min(minWindow, right - left + 1);
int leftNum = a[left];
if (duplicates.contains(leftNum)) {
windowCount.put(leftNum, windowCount.get(leftNum) - 1);
if (windowCount.get(leftNum) < total.get(leftNum) - 1) {
satisfied--;
}
}
left++;
}
}
// 最大剩余长度 = 原长度 - 最小移除窗口长度
System.out.println(n - minWindow);
}
}
复杂度分析
- 时间复杂度:O(n),其中n是数组长度。我们只需要遍历数组两次(统计频率和滑动窗口)
- 空间复杂度:O(n),用于存储元素频率和滑动窗口计数
算法特点
- 高效统计:使用哈希表快速统计元素频率
- 滑动窗口优化:通过滑动窗口技术高效找到最优解
- 边界处理:正确处理无重复元素的特殊情况
- 简洁高效:代码简洁,处理大规模数据效率高
该解法能够高效处理题目给定的最大数据规模(n ≤ 10^5),确保在合理时间内完成计算。