LintCode余弦相似度

来源:互联网 发布:淘宝硅胶充气娃娃 编辑:程序博客网 时间:2024/06/04 19:58

题目描述:给你两个相同大小的向量 A B,求出他们的余弦相似度;返回 2.0000 如果余弦相似不合法 (比如 A = [0] B = [0]).
样例:给出 A = [1, 2, 3], B = [2, 3 ,4]. 返回 0.9926,给出 A = [0], B = [0]. 返回 2.0000
已经给出了公式,直接按着公式计算就行。
可以参考https://yq.aliyun.com/articles/3760 此链接中解释说明

public 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!=B.length)return 2.0000;        double fenzi=0,fenmu1=0,fenmu2=0;        for(int i=0;i<A.length;i++){            fenzi+=A[i]*B[i];            fenmu1+=A[i]*A[i];            fenmu2+=B[i]*B[i];        }        double fenmu = Math.sqrt(fenmu1)*Math.sqrt(fenmu2);        if(fenmu==0)return 2.0000;        double result = fenzi/fenmu;        return result;    }}
原创粉丝点击