如何按自定义模式对字符串数组进行排序

How to sort an array of strings by a custom pattern

本文关键字：数组 排序 字符串 何按 自定义 模式      更新时间：2023-09-26

我有几个数组像:

var arr1 = ['A', 'B', 'C', 'D'], 
    arr2 = ['A', 'C', 'D'],
    arr3 = ['B', 'D'];

我现在希望它们按照自定义模式排序，例如['D', 'C', 'A', 'B']。我正在使用lodash并寻找一种智能的方式来做到这一点。

您可以使用对象来有效地将值映射到可排序的值:

var arr1 = [ "A", "B", "C", "D" ];
var map = { D: 1, C: 2, A: 3, B: 4 };
arr1.sort(function(x, y){ return map[x] - map[y]; });
// show result in snippet
document.write(JSON.stringify(arr1));

与Guffa的解决方案的基本思想相同，但使用了可重用的API: 

var makeSorter = function(pattern) {
    var map = _.mapValues(_.invert(pattern), Number);
    return function(vals) {
        return vals.sort(function(a, b) {
            return map[a] - map[b];
        });
    };
};

var mySorter = makeSorter(['D', 'C', 'A', 'B']);
mySorter(['A', 'B', 'C', 'D']); //=> ["D", "C", "A", "B"]
mySorter(['A', 'C', 'D']); //=> ["D", "C", "A"]

如果你的数据中可能有一些值不在你的模式中，那么你可能需要一些稍微复杂一点的东西:

var makeSorter = function(pattern) {
    var map = _.mapValues(_.invert(pattern), function(s) {return Number(s) + 1;});
    return function(vals) {
        return vals.sort(function(a, b) {
            return (map[a] || Infinity) - (map[b] || Infinity);
        });
    };
};
mySorter(['A', 'X', 'B', 'C', 'Q']); //=> ["C", "A", "B", "X", "Q"]

还有，ES6的箭头函数是多么的好。这相当于ES6中的第一个版本:

var makeSorter = pattern => {
    let map = _.mapValues(_.invert(pattern), Number);
    return vals => vals.sort((a, b) => map[a] - map[b]);
};

var pattern = ['D', 'C', 'A', 'B'];
function sortArray (array, pattern) {
    return array.sort(function (a, b) {
        if (pattern.indexOf(a) === pattern.indexOf(b)) {
            return 0;
        } else {
            return pattern.indexOf(a) > pattern.indexOf(b) ? 1 : -1;
        }
    });
}
// use like:
var sortedArray = sortArray(['A', 'C', 'D'], pattern);
相关文章：