c_cpp 将元素分组为k，然后对数组进行排序

Posted 2021-05-14

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

bool cmp (pair <string, int> a, pair <string, int> b) {
    string c= a.first, d= b.first;
    if (c.length() < d.length())
        return 1;
    else if (c.length() > d.length())
        return 0;
    else
        return c<d;
}
int main() {
    int n, k;
    cin>>n>>k;
    vector <int> a(n);
    for (int i=0;i<n;i++)
        cin>> a[i];

    vector < pair <string, int> > v;//to store the grouped elements and starting index of group.
    int i=0;
    while (i<n) {// join the numbers in groups of k elements.
        int count= 0;
        string temp= "";
        while (count<k && i<n) {
            temp+= to_string(a[i]);
            i++;
            count++;
        }
        v.push_back(make_pair(temp, i-k));
    }
    sort (v.begin(), v.end(), cmp);//sort the newly formed numbers

    vector <int> b(n);
    i=0;
    int j=0;

    while (i<v.size()) {
        int t= v[i].second, count= 0;
        while (t<n && count<k) {
            b[j]= a[t];
            t++;
            j++;
            count++;
        }
        i++;
    }
    for(int i=0;i<n;i++)
        cout<<b[i]<<" ";
}