先说两个函数
unique函数:
去重函数
使用方法:unique (首地址,尾地址);
功能:去除相邻的重复元素(只保留一个),并把重复的元素放在最后;
unique 是返回去重后的尾地址;
lower_bound() 函数,在前闭后开区间进行二分查找
lower_bound() 是返回>=val 的位置,当所有元素都小于val,返回last位置;
使用方法:lower_bound(首地址,尾地址,待查找元素val);
一般情况下,我们离散化的话就是开一个结构体:
struct asd{ int val; //存入值 int id; //存入位置 }; asd q[N];
然后按照val从小到大排序一下;
然后取一下就好了;
for(int i=1;i<=n;i++) arr[q[i].id]=i;
但是这样有一个点就是,如果是重复元素呢?
比如1 2 3 3 4
按照这样离散化以后就会变成:1 2 3 4 5
然而我们可能有时候会需要重复元素离散化后的值是一样的,比如上述结果我是需要:1 2 3 3 4
下面介绍一个这个离散化:
先将输入元素存好,并再开一个vector(就叫xs吧)存入,然后将这个xs排序一下,然后利用unique函数去重,最后for一遍,每次利用Lower_bound()查询到>=arr[i]的位置,
复杂度O(nlogn);
#include <bits/stdc++.h> using namespace std; const int N=5e5+10; int n; int arr[N]; vector<int>xs; void print() { for(int i=1;i<=n;i++) printf("%d ",arr[i]); puts(""); } int main() { scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&arr[i]); xs.push_back(arr[i]); } sort(xs.begin(),xs.end()); auto e=unique(xs.begin(),xs.end()); for(int i=1;i<=n;i++) arr[i]=lower_bound(xs.begin(),e,arr[i])-xs.begin()+1; print(); return 0; }