c_cpp 按设置位的降序对数组进行排序

Posted 2021-05-14

#include <bits/stdc++.h>
using namespace std;

// #BitManipulation
// Sort given array in descending order of set bits
// ex: 11->1011 -> 3 set bits

int calc(int a){
    int setbits=0;
    while(a>0){
        if(a%2==1){
            setbits++;
        }
        a=a/2;
    }
    return setbits;
}
int maxIndex(vector<int> a,int l,int r){
    int max=INT_MIN;
    int maxInd;
    for(int i=l;i<=r;i++){
        if(a[i]>max){
            max=a[i];
            maxInd=i;
        }
    }
    return maxInd;
}
void swap(int &a,int &b){
    int t=a;
    a=b;
    b=t;
}
void func(vector<int> &a){
    int n=a.size();
    vector<int> setbits(n);
    for(int i=0;i<n;i++){
        setbits[i]=calc(a[i]);
    }
    for(int i=0;i<n;i++){
        int maxInd=maxIndex(setbits,0,n-1-i);
        swap(setbits[maxInd],setbits[n-1-i]);
        swap(a[maxInd],a[n-1-i]);
    }
}

int main() {
	int t;
	cin>>t;
	while(t--){
	    int n;
	    cin>>n;
	    vector<int> a(n);
	    for(int i=0;i<n;i++){
	        cin>>a[i];
	    }
	    func(a);
	    for(int i=0;i<n;i++){
	        cout<<a[i]<<" ";
	    }
	    cout<<endl;
	}
	return 0;
}