蓝桥杯《修改数组》超详细

学习算法的小菜鸡

已于 2022-03-28 18:53:01 修改

阅读量1.5k

点赞数 16

文章标签：蓝桥杯算法思维 set 二分

于 2022-03-28 18:45:59 首次发布

本文链接：https://blog.csdn.net/weixin_52314777/article/details/123795839

版权

文章目录

🚀题目描述

给定一个长度为 N 的数组 $A=[A_1,A_2,⋅⋅⋅A_N]$ ，数组中有可能有重复出现的整数。
现在小明要按以下方法将其修改为没有重复整数的数组。
小明会依次修改 $A_2,A_3,⋅⋅⋅,A_N$ 。
当修改 $A i$ 时，小明会检查 $A_i$ 是否在 $A_1∼A_{i−1}$ 中出现过。
如果出现过，则小明会给 $A_i$ 加上 $1$ ；如果新的 $A_i$ 仍在之前出现过，小明会持续给 $A_i$ 加 $1$ ，直到 $A_i$ 没有在 $A_1∼A_{i−1}$ 中出现过。
当 $A_N$ 也经过上述修改之后，显然 $A$ 数组中就没有重复的整数了。
现在给定初始的 $A$ 数组，请你计算出最终的 $A$ 数组。

🚂解题报告

ᾀ暴力（最直接的思路）

🎀做法描述

这道题首先一个比较明确的做法就是说标记每一个数字有没有被使用过。然后每次暴力的往后枚举，找到第一个没有用过的数，输出并标记这个数。很明显，这个做法的复杂度是 $O(n^2)$ 的。

😦代码

#include<iostream>

using namespace std;

const int N=1e6+10;
int st[N],n;

int main() {
    cin>>n;
    for(int i=0;i<n;i++) {
        int a;
        scanf("%d",&a);
        while(st[a]) a++;
        cout<<a<<' ';
        st[a]=1;
    }
}

奇奇怪怪蓝桥杯居然还有90emmm

ᾐ暴力优化

🔠做法描述

观察一下上面暴力做法时间都耗在哪了？哎对，就是查找下一个没用过的数。既然查找费时，那我们就想到用二分来优化。二分要满足二段性，那我们就给他创造一个二段性嘛。我们把用过的数字大一个标记，那么就有：
在这里插入图片描述
然后因为用到了”区间和“这个东西，我们很容易想到用树状数组来做。

🤠代码

#include <iostream>

using namespace std;

const int N=2e6+10;
int q[N];

int lowbit(int x){
    return x&(-x);
}

void add(int x,int v){
    for(int i=x;i<N;i+=lowbit(i)){
        q[i]+=v;
    }
    return ;
}

int query(int x){
    int sum=0;
    for(int i=x;i;i-=lowbit(i)){
        sum+=q[i];
    }
    return sum;
}

int main(){
    int n; 
    cin>>n;
    for(int i=1;i<=n;i++){
        int x; cin>>x;
        int l=x,r=1e6+10;
        while(l<r){
            int mid=(l+r)>>1;
            if(query(mid)-query(x-1)<(mid-x+1)){
                r=mid;
            }
            else l=mid+1;
        }
        add(r,1);
        cout<<r<<" ";
    }
    return 0;
}

不是很满意这个时间，继续干

ὐ集合

🐠做法描述

我们上一个做法中就说明过了他就是区间性质，我们把所有没用过的数字看成一根线段（区间），然后依次查询每一个数字，然后每一次遍历到 $q_i$ 我们就找到他后面最近的一条线段，然后输出左端点，然后区间左端点右移。或者还有一种情况就是他正好在线段上，也就是说这个数本身就没被用到过，我们就可以把线段直接拆成两条（从中间切断）。
在这里插入图片描述

🦀代码

#include <iostream>
#include <set>

using namespace std;

typedef pair<int, int > PII;


void run() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
}

const int N = 5e5 + 10;
int cnt[N];

set<PII>q;

void insert(int l, int r) {
    if (l > r) return;
    q.insert(make_pair(r, l));
}

int main() {
    int n;
    cin >> n;
    q.insert(make_pair(2e9 + 10, 0));
    for (int i = 0; i < n; i++) cin >> cnt[i];
    for (int i = 0; i < n; i++) {
        set<PII>::iterator it;
        it = q.lower_bound(make_pair(cnt[i], 0));
        if (it->second <= cnt[i]) {
            cout << cnt[i] << ' ';
            insert(it->second, cnt[i] - 1);
            insert(cnt[i] + 1, it->first);
            q.erase(it);
        }
        else {
            cout << it->second << ' ';
            insert(it->second + 1, it->first);
            q.erase(it);
        }
    }
}