Java 带index信息排序(包含集合框架的使用)

在进行排序的时候,我们一般得到的是排序后的结果。但是我们有时候不仅需要排序完的结果,还需要排序后对应原来序列的索引。毕图matlab中

[result, index] = sort(a)

result是排序后的结果,index为排序后对应得顺序索引
比如数组a = [1, 6, 3, 4],排序后result=[1, 3, 4, 6], index = [1, 3, 4, 2],即位置信息,6以前是第二个元素,现在排到第4个。
像Java、Python等基本都没有提供这个机制,对于Java,我们可以借助其集合框架来解决这个问题。

不允许有相同元素

使用TreeMap进行排序

 public static int[] sortAndOriginalIndex(Float[] arr) {
    int[] sortedIndex = new int[arr.length];
     TreeMap<Float, Integer> map = new TreeMap<Float, Integer>();
        for (int i = 0; i < arr.length; i++) {
       map.put(arr[i], i); // 将arr的“值-索引”关系存入Map集合
     }
  // System.out.println(map); // 打印集合看看
//  System.out.println("打印格式 -- Value:Index");
  // 使用Entry方式打印Map中的元素
  int n=0;
  Iterator<Map.Entry<Float, Integer>> it = map.entrySet().iterator();
  while (it.hasNext()) {
   Map.Entry<Float, Integer> me = it.next();
//  System.out.print(me.getKey() + ":" + me.getValue() + "\t");
      sortedIndex[n++] = me.getValue();
  }
  return sortedIndex;
 }

允许有相同元素

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.TreeMap;
public class TreeMapSort {
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Map<Integer, String> map = new TreeMap<Integer, String>();
        map.put(1, "ddddd");
        map.put(2, "bbbbb");
        map.put(3, "aaaaa");
        map.put(4, "ccccc");
        map.put(5, "aaaaa");

        //这里将map.entrySet()转换成list
        List<Map.Entry<Integer,String>> list = new ArrayList<Map.Entry<Integer,String>>(map.entrySet());
        //然后通过比较器来实现排序
        Collections.sort(list,new Comparator<Map.Entry<Integer,String>>() {
            //升序排序
            public int compare(Entry<Integer, String> o1,
                    Entry<Integer, String> o2) {
                return o1.getValue().compareTo(o2.getValue());
            }

        });

        for(Map.Entry<Integer,String> mapping:list){ 
               System.out.println(mapping.getKey()+":"+mapping.getValue()); 
          } 
    }
}
####
3:aaaaa
5:aaaaa
2:bbbbb
4:ccccc
1:ddddd
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值