欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

疑难杂题笔记——修改数组

程序员文章站 2023-12-27 08:28:39
...

题目

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

思路

疑难杂题笔记——修改数组

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>

using namespace std;

const int N = 1100010;

int p[N];

int find(int x)
{
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main()
{
    int n;
    cin >> n;
    for(int i = 1;i < N;i++)  p[i] = i;  //表示都没有被用过,全都是指向自身的环
    for(int i = 1;i <= n;i++)
    {
        int x;
        cin >> x;
        x = find(x);  //找出这个链往后数的第一个没有被用过的点
        cout << x << ' ';
        p[x] = x + 1;  //向后移动一位
    }
    return 0;
}

代码2

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>

using namespace std;

const int N = 1000010;

int st[N];

int main()
{
    int n;
    cin >> n;
    for(int i = 0;i < n;i++)
    {
        int a;
        cin >> a;
        while(st[a])
        {
            a = st[a]++;
        }
        st[a] = a;
        cout << a << ' '; 
    }
    return 0;
}
相关标签: 算法学习 算法

上一篇:

下一篇: