arrays - 以另一个数组作为 Chapel 中的索引对数组进行排序

Question

我正在查看 Chapel 中的Sort运算符并尝试对两个数组进行排序。

const a = [2.2, 3.3, 1.1, 4.4];
const b = [7, 10, 23, 1];

现在我想要一个迭代器或新数组，它b按a. 那是

var output = [1, 10, 7, 23];

我如何使用比较器（或者，真的，任何方法）来做到这一点？

Answer 1

您可以通过将数组压缩在一起并使用比较器对它们进行排序来完成此操作，比较器定义了要排序的数组（压缩表达式的元组索引）和顺序（升序或降序）。有几种不同的方法可以实现这一点，其中一种如下所示：

use Sort;

/* Reverse comparator sorting by first element of tuple */
record Cmp {
  proc key(a) return a[1];
}

// Create a comparator instance with reversed order
var cmp: ReverseComparator(Cmp);

/* Iterate over elements of array2 reverse-sorted by values of array1 */
iter zipSort(array1, array2) {

  // Concatenate the arrays into an array of tuples
  const AB = [ab in zip(array1, array2)] ab;

  // Sort array with comparator created above
  for ab in AB.sorted(comparator=cmp) {
    yield ab[2];
  }
}

proc main() {
  const A = [2.2, 3.3, 1.1, 4.4];
  const B = [7, 10, 23, 1];

  // Iterator
  for b in zipSort(A, B) {
    writeln(b);
  }

  // New array
  var output = zipSort(A, B);
  writeln(output);
}

我们可以根据您的具体情况稍微简化一下。由于默认情况下元组按第一个元素排序，因此我们实际上不需要创建比较器，但仍需要使用模块中reverseComparator提供的反转顺序Sort：

use Sort;

/* Iterate over elements of array2 reverse-sorted by values of array1 */
iter zipSort(array1, array2) {

  // Concatenate the arrays into an array of tuples
  const AB = [ab in zip(array1, array2)] ab;

  // Sort array with comparator created above
  for ab in AB.sorted(comparator=reverseComparator) {
    yield ab[2];
  }
}