[蓝桥杯2019初赛]修改数组

修改数组

题目描述
给定一个长度为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(1<=N<=100000)
第二行包含N个整数A1,A2,…,AN(1<=Ai<=1000000)
输出
输出N个整数,依次是最终的A1,A2,…,AN
样例输入
5
2 1 1 3 4
样例输出
2 1 3 4 5

思路:这个题暴力肯定不行,O(n^2)的算法一定拿不了满分,Hash也不行,然后我就不会了,后来看了倪文迪老师的博客后惊奇的发现居然可以用并查集做,以前对并查集的认识在最小生成树,没想到还能这么用,把值相同的数字看成一个集合,然后当它出现一次就把它接到当前的根+1的位置,然后直接找根赋值就行了(这我是没想到的)

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,a[N],pre[N];
int find_(int x)
{
    if(pre[x]==x)return pre[x];
    int fn=find_(pre[x]);
    return pre[x]=fn;
}
void merge_(int x,int y)
{
    int tmp1,tmp2;
    tmp1=find_(x);
    tmp2=find_(y);
    if(tmp1!=tmp2)pre[tmp1]=tmp2;
}
int main()
{
    for(int i=1;i<=N;i++)pre[i]=i;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        int root=find_(a[i]);
        a[i]=root;
        merge_(a[i],a[i]+1);
    }
    for(int i=1;i<=n;i++)printf("%d ",a[i]);
    return 0;
}

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值