蓝桥杯-试题H:修改数组

Posted 2021-11-11 nuist__NJUPT

试题H:修改数组

方法1：辅助数组遍历法

import java.util.Scanner;

public class Main1 {
    public static void f(int [] arr){
        boolean flag = false ;
        int [] helper = new int [arr.length] ; //辅助数组
        helper[0] = arr[0] ;
        for(int i=1; i<arr.length; i++){
            for(int j=0; j<i; j++){
                if(arr[i] == helper[j]){
                    arr[i] += 1 ;
                    helper[i] = arr[i] ;
                    j = -1 ;
                    flag = true ;
                }
            }
            if(flag == false){
                helper[i] = arr[i] ;
            }else{
                flag = false ;
            }
        }
        for(int i=0; i<helper.length; i++){
            System.out.print(helper[i] + " ");
        }
    }
    public static void main(String[] args){
        Scanner input = new Scanner(System.in) ;
        int n = input.nextInt() ;
        int [] arr = new int [n] ;
        for(int i=0; i<n; i++){
            arr[i] = input.nextInt() ;
        }
        f(arr) ;
    }
}

方法2：并查集法

/**
 * 比如2 1 1 3 4，首先输出就是2，然后并查集使得数组元素f[2]由本来的2，+1变成了3，
 * 也就是之后再碰到2会直接输出3。后面都是同理。
 */

import java.util.Scanner;

public class Main2 {
    static int [] f = new int [1000001] ;
    public static int find(int x){
        //对代表元素的查询，一层一层的查询，直至访问到根节点,判断两个元素是否是否属于同一集合，只需要判断他们是否有相同的根节点
        if(x == f[x]){
            return x ;
        }else{
         f[x] = find(f[x]) ; //路径压缩，将沿途的每个父节点都设置为根节点
         return f[x] ;
        }
    }
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in) ;
        int n = input.nextInt() ;
        int [] data = new int [n] ;
        for(int i=0; i<n; i++){
            data[i] = input.nextInt() ;
        }
        for(int i=1; i<f.length; i++){ //初始化，用f[]数组来存储每个元素的父节点
            f[i] = i ;
        }
        for(int i=0; i<data.length; i++){
            int k = find(data[i]) ;//查询代表元素
            data[i] = k ;
            f[data[i]] = find(data[i] + 1) ; //合并，将前者的父节点设为后者
        }
        for(int j=0; j<data.length; j++){
            System.out.print(data[j] + " ");
        }
    }
}