余弦相似度
题目
wiki链接: Cosine Similarity
这里给出公式:
给你两个相同大小的向量 A B,求出他们的余弦相似度
返回 2.0000 如果余弦相似不合法 (比如 A = [0] B = [0]).样例
给出 A = [1, 2, 3], B = [2, 3 ,4].
返回 0.9926.
给出 A = [0], B = [0].
返回 2.0000题解
class Solution {
/**
* @param A: An integer array.
* @param B: An integer array.
* @return: Cosine similarity.
*/
public double cosineSimilarity(int[] A, int[] B) {
double bottom_left = 0;
double bottom_right = 0;
double top = 0;
int n = A.length;
for (int i = 0;i < n; i++)
{
bottom_left += A[i] * A[i];
bottom_right += B[i] * B[i];
top += A[i] * B[i];
}
double temp = Math.sqrt(bottom_left) * Math.sqrt(bottom_right);
return temp == 0 ? 2.0000 : top / temp;
}
}
Last Update 2016.9.17