c++算法之离散化例题

离散化基础2

题目描述

给定 n 个元素的数列，将相同的数据离散化为一个数据（去重），即把 {4000,201,11,45,11}{4000,201,11,45,11} 离散化为 {4,3,1,2,1}{4,3,1,2,1}。

输入格式

第一行一个整数 (1≤m≤105)n(1≤n≤105)，为元素的个数。

第二行 n 个用空格隔天的整数 a[i](−109≤a[i]≤109)ai​(−109≤ai​≤109)，为数列中的元素。

输出格式

一行若干个整数，为离散化的结果，用一个空格隔开。

样例 #1

样例输入 #1

5
4000 201 11 45 11

样例输出 #1

4 3 1 2 1

#include<iostream>
#include<algorithm>
using namespace std;
int a[101100],b[101100];
int main(){
    int n;
    cin >> n;
    for(int i=1;i<=n;i++){
        cin >> a[i];//未排序 
        b[i]=a[i];//副本 
    }
    sort(b+1,b+n+1);
    int cnt=unique(b+1,b+n+1)-(b+1);
    for(int i=1;i<=n;i++){
        //二分查找函数:>=a[i]的地址，“a[i]-b”就可以返回编号 
        a[i]=lower_bound(b+1,b+cnt+1,a[i])-b;
        printf("%d ",a[i]);
    } 
    return 0;
}