题目

wiki链接: Cosine Similarity
这里给出公式:

/media/problem/cosine-similarity.png

样例
给出 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
                int ab=0;
        if(sumArray(A) == 0 || sumArray(B) == 0)
        {
            return 2.00000;
        }
        for(int i=0;i<A.length;i++)
        {
            ab += A[i] * B[i];
        }
        return ab/Math.sqrt(sumArray(A))/Math.sqrt(sumArray(B));
    }
    
        public double sumArray(int[] A)
    {
        int sum = 0;
        for(int i=0;i<A.length;i++)
        {
            sum += A[i] * A[i];
        }
        return sum;
    }
}