LeetCode #2948

Make Lexicographically Smallest Array by Swapping Elements

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
class Solution {
public:
    vector<int> lexicographicallySmallestArray(vector<int>& nums, int limit) {
        int N = (int)nums.size();
        vector<int> idx(N); iota(idx.begin(), idx.end(), 0);
        sort(idx.begin(), idx.end(), [&](int u, int v) -> bool { return nums[u] < nums[v]; });
        vector<vector<int>> G;
        G.emplace_back(); G.back().push_back(idx[0]);
        for(int i = 1; i < N; ++i) {
            if(nums[idx[i]] - nums[G.back().back()] > limit) G.emplace_back();
            G.back().push_back(idx[i]);
        }
        vector<vector<int>> seq = G;
        vector<int> answer(N);
        for(int i = 0; i < (int)seq.size(); ++i) {
            sort(seq[i].begin(), seq[i].end());
            for(int j = 0; j < (int)seq[i].size(); ++j) answer[seq[i][j]] = nums[G[i][j]];
        }
        return answer;
    }
};

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.