代码如下:
int num[maxn]; //原数据数组
int tp[maxn]; // 中间数组
int ans[maxn]; //离散化后数组
int n; //数据数量
map<int, int> mp; //原数组与离散化后数据的映射关系
int id[maxn]; //离散化后的数据对原数据的映射
for (int i = 0; i < n; ++i) {
tp[i] = num[i];
}
sort(tp, tp + n);
int m = unique(tp, tp + n) - tp; // 对数组去重,m 为去重后数组中元素个数
for (int i = 0; i < m; ++i) {
mp[tp[i]] = i + 1; //建立原数据与离散化后数据的映射
id[i+1] = tp[i];
}
for (int i = 0; i < n; ++i) {
ans[i] = mp[num[i]];
}
数据离散化的模板如下:
#include <iostream>
#include <algorithm>
#include <map>
using namespace std;
const int maxn=10010;
int num[maxn];
int temp[maxn];
int n;
map<int,int>mp;
int main() {
cin>>n;
for(int i=0;i<n;i++){
cin>>num[i];
temp[i]=num[i];
}
sort(temp,temp+n);
int m=unique(temp,temp+n)-temp;
for(int i=0;i<m;i++){
mp[temp[i]]=i+1;
}
for(int i=0;i<n;i++){
cout<<mp[num[i]]<<" ";
}
return 0;
}