时间限制: 1.0s 内存限制: 256.0MB 本题总分:20 分
【问题描述】
给定一个长度为 N 的数组 A = [A1, A2, · · · AN],数组中有可能有重复出现的整数。
现在小明要按以下方法将其修改为没有重复整数的数组。小明会依次修改
A2, A3, · · · , AN。
当修改 Ai 时,小明会检查 Ai 是否在 A1 ~ Ai-1 中出现过。如果出现过,则小明会给 Ai 加上 1 ;如果新的 Ai 仍在之前出现过,小明会持续给 Ai 加 1 ,直到 Ai 没有在 A1 ~ Ai-1 中出现过。
当 AN 也经过上述修改之后,显然 A 数组中就没有重复的整数了。现在给定初始的 A 数组,请你计算出最终的 A 数组。
【输入格式】
第一行包含一个整数 N。
第二行包含 N 个整数 A1, A2, · · · , AN 。
【输出格式】
输出 N 个整数,依次是最终的 A1, A2, · · · , AN。
【样例输入】
5
2 1 1 3 4
【样例输出】
2 1 3 4 5
【评测用例规模与约定】
对于 80% 的评测用例,1 ≤ N ≤ 10000。
对于所有评测用例,1 ≤ N ≤ 100000,1 ≤ Ai ≤ 1000000。
思路:
首先乍一看这题巨简单,之间放在集合里判重就ok了,但是分析一下时间限制,应该是过不了的
这里推荐一个方法可以在比赛中检测自己代码是否超时
long begin= new Date().getTime();
----代码块----
long end= new Date().getTime();
System.out.println(end-begin);
这样就可以判断是否超时
下面展示一下我这个菜鸡的 超时代码
package competion2016;
import java.util.ArrayList;
import java.util.Date;
import java.util.Scanner;
public class 修改数组 {
static int n;
static int[] arr;
static ArrayList<Integer> list=new ArrayList<>();
public static void main(String[] args) {
// TODO 自动生成的方法存根
long startTime = new Date().getTime();
Scanner sc =new Scanner(System.in);
n=Integer.parseInt(sc.nextLine());
arr=new int[n];
for (int i = 0; i < arr.length; i++) {
arr[i]=sc.nextInt();
}
list.add(arr[0]);//先加入第一个元素
for (int i = 1; i < n; i++) {
while (list.contains(arr[i])) {
arr[i]++;
}//存在不加
list.add(arr[i]);//不存在直接加
}
for (int i = 0; i < list.size(); i++) {
System.out.print(list.get(i));
}
long endTime = new Date().getTime();
System.out.println("本程序运行 " + (endTime - startTime)
+ " 毫秒完成。" );
}
}
右边的就是超时的结果大于1s,左边的是优化后的代码不超时
下面使用并查集优化
思路:每次赋给数组的值是当前元素的祖先
每次赋值结束后,将该元素的祖先设置为比自身大1,这样再遇到该元素时,赋给数组该元素的祖先就一定和它不同。且比这条祖先链中最大的数还大1;
不会思路一定要懂得调试
(fa数组有点大,建议调试时用运行光标调试法)
import org.omg.CosNaming.BindingIterator;
import java.util.ArrayList;
import java.util.Date;
import java.util.Scanner;
public class ad {
static int[] fa=new int[2000000];
public static void main(String[] args) {
long begin= new Date().getTime();
Scanner scanner=new Scanner(System.in);
int n=scanner.nextInt();
int[] arr=new int[n];
for (int i = 0; i < n; i++) {
arr[i]=scanner.nextInt();
}
//初始化f
for (int i = 1; i < fa.length; i++) {
fa[i]=i;
}
for (int i = 0; i < arr.length; i++) {
int num=find(arr[i]);//将arr[i]的祖先返回
arr[i]=num;//赋值
fa[arr[i]]=find(arr[i]+1);//将添加的元素的祖先设置为比自身大1,实现了每次赋给arr的祖先值都不存在且比最大大1,这里的合并过程其实就实现了不存在
}
for (int i = 0; i < arr.length; i++) {
System.out.print(arr[i]+" ");
}
long end= new Date().getTime();
System.out.println(end-begin);
}
//查找,返回的祖先一定是不存在于序列的
public static int find(int x) {
return fa[x]=fa[x]==x?x:find(fa[x]);
}
}