余弦相似度

wiki链接: Cosine Similarity

这里给出公式:

/media/problem/cosine-similarity.png

给你两个相同大小的向量 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) {
        // write your code here
        if (A.length == 0 || B.length == 0 || A.length != B.length){
            return 2.0000;
        }
        int a = 0;
        int b = 0;
        int c = 0;
        for (int i = 0; i < A.length; i++){
            a += A[i] * A[i];
            b += B[i] * B[i];
            c += A[i] * B[i];
        }
        if (a == 0 || b == 0){
            return 2.0000;
        }
       return c/( Math.sqrt(a) * Math.sqrt(b));
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值