445.Cosine Similarity-余弦相似度（容易题）

三色旗飞扬

于 2016-09-17 01:32:35 发布

阅读量845

点赞数

CC 4.0 BY-SA版权

分类专栏： LintCode笔记文章标签： LintCode

本文链接：https://blog.youkuaiyun.com/Tri_Color_Flag/article/details/52560233

LintCode笔记专栏收录该内容

259 篇文章

订阅专栏

本文介绍了一种衡量向量间相似性的方法——余弦相似度，并给出了具体的计算公式及实现代码。当输入向量全部为零时，定义余弦相似度为2.0000。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

余弦相似度

题目

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;
    }
}