前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >脚撕LeetCode(1356)Easy

脚撕LeetCode(1356)Easy

作者头像
JathonKatu
发布2022-01-18 08:25:11
1590
发布2022-01-18 08:25:11
举报
文章被收录于专栏:JathonKatu

题目地址:https://leetcode-cn.com/problems/sort-integers-by-the-number-of-1-bits/

给你一个整数数组arr。 请你将数组中的元素按照其二进制表示中数字 1 的数目升序排序。 如果存在多个数字二进制中1的数目相同,则必须将它们按照数值大小升序排列。 请你返回排序后的数组。

代码语言:javascript
复制
示例 1: 
输入:arr = [0,1,2,3,4,5,6,7,8] 
输出:[0,1,2,4,8,3,5,6,7] 
解释:[0] 是唯一一个有 0 个 1 的数。[1,2,4,8] 都有 1 个 1 。[3,5,6] 有 2 个 1 。[7] 有 3 个 1 。按照 1 的个数排序得到的结果数组为 [0,1,2,4,8,3,5,6,7] 
示例 2: 
输入:arr = [1024,512,256,128,64,32,16,8,4,2,1] 
输出:[1,2,4,8,16,32,64,128,256,512,1024] 
解释:数组中所有整数二进制下都只有 1 个 1 ,所以你需要按照数值大小将它们排序。 
示例 3:
输入:arr = [10000,10000] 
输出:[10000,10000] 
示例 4: 
输入:arr = [2,3,5,7,11,13,17,19] 
输出:[2,3,5,17,7,11,13,19] 
示例 5: 
输入:arr = [10,100,1000,10000] 
输出:[10,100,10000,1000]  
提示: 
1 <= arr.length <= 500 
0 <= arr[i] <= 10^4

这道题的思路是:根据数组中,每个元素的二进制表达中,1的个数排序,少的排在前面。如果1的个数一样就根据数字大小排序。

一、爆破法

爆破法的思路很简单,先sort,用map保存,key为二进制中1的个数,value为数组,存放数字。

最后遍历map即可(提示中给出0 <= arr[i] <= 10^4,所以只要遍历到14位即可)

执行结果如下:

77 / 77 个通过测试用例

状态:通过

执行用时: 11 ms

内存消耗: 38.8 MB

代码语言:javascript
复制
public static int[] sortByBitsMe(int[] arr) {
    Map<Integer, ArrayList<Integer>> map = new HashMap<>(16, 1);
    Arrays.sort(arr);
    int count = 0;
    for (int i : arr) {
        count = binaryHaveOneNum(i);
        if(map.get(count) == null) {
            map.put(count,new ArrayList<>());
        }
        map.get(count).add(i);
    }
    int end = -1;
    for (int i = 0; i <= 14; i++) {
        ArrayList<Integer> itemList = map.get(i);
        if(null == itemList) continue;
        for (Integer integer : itemList) {
            arr[++end] = integer;
        }
    }
    return arr;
}

public static int binaryHaveOneNum(int num) {
    int ans = 0;
    for (int i = 0; i < 14; i++) {
        if ((num >>> i & 1) == 1) ans++;
    }
    return ans;
}

这个办法很爆破,而且也很慢,但是自己毕竟经历了一天想不出来好的办法,今天只能用上愚蠢的爆破法。

下面看看评论区大佬的做法

二、位运算排序法

大佬的做法是,首先在一个高位存储数字的二进制中1的个数。

然后排序(如果1个个数相同,那么数字大小不同也能准确排序,减少了一次排序)

最后再将低位取出,重新赋值即可。

执行结果如下:

77 / 77 个通过测试用例

状态:通过

执行用时: 3 ms

内存消耗: 38.6 MB

代码语言:javascript
复制
public int[] sortByBits(int[] arr) {
    for (int i = 0; i < arr.length; i++) {
        arr[i] += (Integer.bitCount(arr[i]) << 16);
    }
    Arrays.sort(arr);
    for (int i = 0; i < arr.length; i++) {
        arr[i] &= 16383;
    }
    return arr;
}

代码量少了很多,执行时间也少了很多,内存占用差不多。果然自己对位运算还是用的不熟练啊

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2021-06-15,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 JathonKatu 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档