image.png
class Solution {
public:
/**
* @param A: An integer array.
* @param B: An integer array.
* @return: Cosine similarity.
*/
//如果vector 中的所有元素都为空，那么返回false
bool is_zero(vector<int> a){
for(auto it=a.begin();it!=a.end();it++){
if(*it!=0){
return true;
}
}
return false;
}
double cosineSimilarity(vector<int> A, vector<int> B) {
// write your code here
if(A.empty()==true){
return 2.0000;
}
if((is_zero(A)==false)&&(is_zero(B)==false)){
return 2.0000;
}
double res1=0;
for(int i=0;i<A.size();i++){
res1=res1+A[i]*B[i];
}
double res2=0;
for(int i=0;i<A.size();i++){
res2=res2+A[i]*A[i];
}
res2=sqrt(res2);
double res3=0;
for(int i=0;i<B.size();i++){
res3=res3+B[i]*B[i];
}
res3=sqrt(res3);
return res1/(res2*res3);
}
};
