Untitled

 avatar
unknown
plain_text
a year ago
601 B
1
Indexable
class Solution {
public:
    vector<vector<int>> divideArray(vector<int>& arr, int k) {
        int size=arr.size()/3, n=arr.size();
        if(n%3 !=0 ) return vector<vector<int>>();

        vector<vector<int>>res(size,vector<int>(3));
        sort(arr.begin(),arr.end());

        int ind=0;
        for(int i=0;i<n;i+=3){
            if(i+2<n && arr[i+2]-arr[i]<=k){
                res[ind]={arr[i],arr[i+1],arr[i+2]};
                ind++;
            }
            else if(arr[i+2]-arr[i]>k){
                return vector<vector<int>>();
            }
        }
        return res;  
    }
};
Leave a Comment