Question

我正在寻找一种方法来改善以下算法的性能。给定两个数组X和Y.

对于X的每个元素，找到Y中最大值的索引，该索引不超过X中元素的值。可以安全地假设X和Y单调递增（排序）并且Y（1）是少于X中的每个值。 X通常比Y大得多。

以下为例。

X = [0.2, 1.5, 2.2, 2.5, 3.5, 4.5, 5.5, 5.8, 6.5];
Y = [0.0, 1.0, 3.0, 4.0, 6.0];

我希望输出为

idx = [1, 2, 2, 2, 3, 4, 4, 4, 5]

我提出的最快的方法是下面的函数无法利用列表排序并使用for循环逐步遍历其中一个数组的事实。这提供了一个有效的解决方案，但在我使用此功能的实验中，在分析所需的总共30分钟内花费了将近27分钟。

function idx = matchintervals(X,Y)
  idx = zeros(size(X));
  for i = 1:length(Y)-1
    idx(X >= Y(i) & X < Y(i+1)) = i;
  end
  idx(X >= Y(end)) = length(Y);
end

非常感谢任何帮助。

Answer 1

如果您正在寻找最快的解决方案，它可能最终会成为一个简单的while循环（这样可以利用数组排序的事实）：

X = [0.2, 1.5, 2.2, 2.5, 3.5, 4.5, 5.5, 5.8, 6.5];
Y = [0.0, 1.0, 3.0, 4.0, 6.0];

xIndex = 1;
nX = numel(X);
yIndex = 1;
nY = numel(Y);
index = zeros(size(X))+nY;  % Prefill index with the largest index in Y

while (yIndex < nY) && (xIndex <= nX)
  if X(xIndex) < Y(yIndex+1)
    index(xIndex) = yIndex;
    xIndex = xIndex+1;
  else
    yIndex = yIndex+1;
  end
end

>> index

index =

     1     2     2     2     3     4     4     4     5

如果numel(X)+numel(Y)-1中的多个值大于X中的最大值，则此循环将迭代最多Y次，可能更少。

TIMINGS：我使用评论中的示例数据运行了一些时间。以下是从最快到最慢排序的结果：

X = 1:3:(4e5);
Y = 0:20:(4e5-1);

% My solution from above:
tElapsed =
   0.003005977477718 seconds

% knedlsepp's solution:
tElapsed =
   0.006939387719075 seconds

% Divakar's solution:
tElapsed =
   0.011801273498343 seconds

% H.Muster's solution:
tElapsed =
   4.081793325423575 seconds

Answer 2

单行，但可能比gnovice的解决方案慢：

idx = sum(bsxfun(@ge, X, Y'));

Answer 3

使用sort和少数masks -

%// Concatenate X and Y and find the sorted indices
[sXY,sorted_id] = sort([X Y]);

%// Take care of sorted_id for identical values between X and Y
dup_id = find(diff(sXY)==0);
tmp = sorted_id(dup_id);
sorted_id(dup_id) = sorted_id(dup_id+1);
sorted_id(dup_id+1) = tmp;

%// Mask of Y elements in XY array
maskY = sorted_id>numel(X);

%// Find island lengths of Y elements in concatenated XY array
diff_maskY = diff([false maskY false]);
island_lens = find(diff_maskY ==-1) - find(diff_maskY ==1);

%// Create a mask of double datatype with 1s where Y intervals change
mask_Ys = [ false maskY(1:end-1)];
mask_Ysd = double(mask_Ys(~maskY));

%// Incorporate island lengths to change the 1s by offsetted island lengths
valid = mask_Ysd==1;
mask_Ysd(valid) = mask_Ysd(valid) + island_lens(1:sum(valid)) - 1;

%// Finally perform cumsum to get the output indices
idx = cumsum(mask_Ysd);

Answer 4

我和 Divakar 有类似的想法。这基本上使用稳定X在值Y之后的sort中查找值的插入点。需要对X和Y进行排序才能正常工作！

%// Calculate the entry points
[~,I] = sort([Y,X]);
whereAreXs = I>numel(Y);
idx = find(whereAreXs)-(1:numel(X));

您可以通过以下方式查看X的值以及不超过Y值的X的相应值：

%%// Output:
disp([X;Y(idx)]);

对于X中的每个元素，找到最大的索引，而不是在Y中

4 个答案: