C++犰狳：最近的鄰居

Question

我在C++中使用了Armadillo庫。

假設我有一個n×1列矩陣，它按數字順序排序。例如

mat X; X.randn(100,1); 
mat X_sorted; X_sorted = sort(X); 
cout << X_sorted << endl; 

，並假設我有一個變量

double y = 0.5; 

我想：是找到指數，z的x_sorted，爲此x_sorted(z)是最接近爲y的一種方式。如果是領帶（這在我的情況中並不重要），只需選擇較大的那個。這樣做的

Answer 1

一種方法是這樣的：

int z = as_scalar(sort_index(abs(X_sorted - y)).row(0)); 

請隨意批評這個解決方案，並提出改善。

下面是它的一個示例程序在行動：

int main(int argc, char** argv) 
{ 
using namespace arma; 
using namespace std; 

mat X; X.randn(100,1); 
mat X2; X2.zeros(100,1); 
for(int i=0; i<100; i++){X2(i) = i;} 
mat X_sorted; X_sorted = sort(X); 

mat XX; XX=join_rows(X2,X_sorted); 
cout << XX << endl; 

double y = 0.5; 
int z = as_scalar(sort_index(abs(X_sorted - y)).row(0)); 
mat XX_z; XX_z = XX.row(z); 
cout << XX_z << endl; 

return 0; 
}