ZSort.cpp

//Copyright [2018] [Haibo Yan]
//
//Licensed under the Apache License, Version 2.0 (the "License");
//you may not use this file except in compliance with the License.
//You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
//Unless required by applicable law or agreed to in writing, software
//distributed under the License is distributed on an "AS IS" BASIS,
//WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//See the License for the specific language governing permissions and
//limitations under the License.

//
// Created by haibo on 3/16/18.
//

#include "ZSort.h"

#include <algorithm>

void ZSort::zsort(vector<string> &strings, int k) {
    stable_sort(strings.begin(), strings.end(), [k](string s1, string s2){
        if (s1.size() < k) {
            return false;
        }

        if (s2.size() < k) {
            return true;
        }

        return s1[k - 1] < s2[k - 1];
    });
}