概述
题目地址:
https://www.lintcode.com/problem/longest-super-sequence/description
给定一个长 n n n数组 A A A,要求找到其最长子数组,删掉其一个元素之后别的元素出现次数都一样。返回这个子数组。题目保证 A A A里只含 0 ∼ 3000 0sim 3000 0∼3000。
思路是枚举“别的元素出现次数都一样”这个出现次数 x x x。先统计一下每个数字的出现次数,接着开个最小堆,将出现了的数入堆,出现次数小的优先级高,并且存一下出现次数最多的那个数是谁。我们的目的是,每次从堆中弹出一个数 a a a的时候,如果其出现次数是 c c c,那么我们枚举 x = c − 1 x=c-1 x=c−1和 x = c x=c x=c的情况。假设poll出 a a a之后堆的size是 s s s。 x = c − 1 x=c-1 x=c−1的情况是指, a a a以及堆里的数都重复 c − 1 c-1 c−1次,另外再加上 a a a自己,此时整个LSS的长度是 ( c − 1 ) ( s + 1 ) + 1 (c-1)(s+1)+1 (c−1)(s+1)+1; x = c x=c x=c的情况是, a a a以及堆里的数都重复 c c c次,再加上一个堆里的某个数多一次,此时整个LSS的长度是 c ( s + 1 ) + 1 c(s+1)+1 c(s+1)+1,这里有个必要条件是 A A A里存在某个数出现次数超过 c c c,否则无法存在这个数,例如 A A A里每个数都出现了 2 2 2次,则 c c c是无法枚举为 2 2 2的,因为没有数次数比 2 2 2多。但是此时 c c c仍然可以枚举为 1 1 1。这里还需要注意 A A A里所有数都只出现了 1 1 1次的情况,这个情况不含在上面的讨论中,此时直接返回 A A A自己。代码如下:
import java.util.Arrays;
import java.util.PriorityQueue;
public class Solution {
/**
* @param nums: the input array
* @return: return the longest super sequence
*/
public int[] longestSuperSequence(int[] nums) {
// write your code here
int[] count = new int[3001];
int maxCount = 0, maxCountNum = 0;
// 统计一下每个数出现次数,以及出现次数最多的数是几,及其出现次数
for (int num : nums) {
count[num]++;
if (count[num] > maxCount) {
maxCount = count[num];
maxCountNum = num;
}
}
// 特判一下出现次数最多的元素只出现一次的情况
if (maxCount == 1) {
return nums;
}
// 建堆,把出现过的数入堆,出现次数少的数优先
PriorityQueue<Integer> minHeap = new PriorityQueue<>((x, y) -> Integer.compare(count[x], count[y]));
for (int i = 0; i < count.length; i++) {
if (count[i] > 0) {
minHeap.offer(i);
}
}
// maxLen是可以达到的最长LSS的长度,mostCount是这个LSS去掉一个数后别的数出现次数一样了,这个出现次数
int maxLen = 0, mostCount = 0;
while (!minHeap.isEmpty()) {
int cur = minHeap.poll(), curCount = count[cur];
if ((count[cur] - 1) * (minHeap.size() + 1) + 1 > maxLen) {
maxLen = (count[cur] - 1) * (minHeap.size() + 1) + 1;
mostCount = count[cur] - 1;
}
if (count[cur] * (minHeap.size() + 1) + 1 > maxLen && maxCount > count[cur]) {
maxLen = count[cur] * (minHeap.size() + 1) + 1;
mostCount = count[cur];
}
// 这里可以做个优化,一旦枚举完这个curCount,就把所有出现次数也是curCount的数全poll出来
while (!minHeap.isEmpty() && count[minHeap.peek()] == curCount) {
minHeap.poll();
}
}
int[] res = new int[maxLen], resCount = new int[3001];
int idx = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] == maxCountNum && resCount[maxCountNum] < mostCount + 1) {
res[idx++] = nums[i];
resCount[nums[i]]++;
} else if (nums[i] != maxCountNum && count[nums[i]] >= mostCount && resCount[nums[i]] < mostCount) {
res[idx++] = nums[i];
resCount[nums[i]]++;
}
}
return res;
}
}
时间复杂度 O ( n log n ) O(nlog n) O(nlogn),空间 O ( n ) O(n) O(n)。
最后
以上就是香蕉月光为你收集整理的【Lintcode】1877. Longest Super Sequence题目地址:的全部内容,希望文章能够帮你解决【Lintcode】1877. Longest Super Sequence题目地址:所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复