1242. 修改数组 并查集
Published on 2022-11-17 23:01 in 暂未分类 with 林动

1242. 修改数组 并查集

    给定一个长度为 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。

    数据范围
    1≤N≤105,
    1≤Ai≤106
    输入样例:
    5
    2 1 1 3 4
    输出样例:
    2 1 3 4 5

    这里并查集的父节点就是下一个可能可选的数,如果f[i]=i,那么表示i未被选过,
    那么选它,并修改f[i]=i+1
    
    
    import java.util.*;
    
    public class Main
    {
    	static int N=100005,M=1100005;
    	static int a[]=new int [N];
    	static int f[]=new int [M];
    	static int find(int x){
    		return x==f[x]?x:(f[x]=find(f[x]));
    	}
    	
    	public static void main(String args[])
    	{
    		Scanner sc=new Scanner(System.in);
    		int n=sc.nextInt();
    		for(int i=0;i<n;++i)a[i]=sc.nextInt();
    		for(int i=0;i<M;++i)f[i]=i;
    		for(int i=0;i<n;++i){
    			int x=a[i];
    			x=find(x);
    			a[i]=x;
    			f[x]=x+1;
    		}
    		for(int i=0;i<n;++i)System.out.print(a[i]+" ");
    	}
    }
    
    
    posted @   林动  阅读(14)  评论(0编辑  收藏  举报
    相关博文:
    阅读排行:
    · 地球OL攻略 —— 某应届生求职总结
    · 周边上新:园子的第一款马克杯温暖上架
    · Open-Sora 2.0 重磅开源!
    · 提示词工程——AI应用必不可少的技术
    · .NET周刊【3月第1期 2025-03-02】
    点击右上角即可分享
    微信分享提示