【模板】离散化实现的两种方式

来源:互联网 发布:大话设计模式 java 编辑:程序博客网 时间:2024/05/22 17:47

第一种:利用结构体实现离散化
这种方法的弊端是无法辨别重复元素
例如
1 2 2 3 3
离散化之后是
1 2 3 4 5
使用时应注意

#include <bits/stdc++.h>  using namespace std;const int N=5e5+10;  int n;int b[N];struct node{    int id,val;    bool operator <(const node& a)const    {        return val<a.val;    }}arr[N];int main(){    scanf("%d",&n);    for(int i=1;i<=n;i++)    {        cin>>arr[i].val;        arr[i].id=i;    }    sort(arr+1,arr+n+1);    for(int i=1;i<=n;i++)   b[arr[i].id]=i;    for(int i=1;i<=n;i++)   cout<<b[i]<<" ";    cout<<endl;}

第二种:利用STL中的unique()和 lower_bound() 函数实现
unique函数:
去重函数
使用方法:unique (首地址,尾地址);
功能:去除相邻的重复元素(只保留一个),并把重复的元素放在最后;
unique 是返回去重后的尾地址;
lower_bound() 函数,在前闭后开区间进行二分查找
lower_bound() 是返回>=val 的位置,当所有元素都小于val,返回last位置;
使用方法:lower_bound(首地址,尾地址,待查找元素val);
upper_bound() 函数,在前闭后开区间进行二分查找
upper_bound() 函数是返回>val 的位置,其余与lower_bound()相同

#include <bits/stdc++.h>  using namespace std;  const int N=5e5+10;  int n;  int arr[N];  int b[N];  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]);          b[i]=arr[i];      }      sort(b+1,b+n+1);      int m=unique(b+1,b+n+1)-b-1;      for(int i=1;i<=n;i++)          arr[i]=lower_bound(b+1,b+m,arr[i])-b;      print();      return 0;  }
原创粉丝点击