-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathday7_1
54 lines (48 loc) · 1.38 KB
/
day7_1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
//In this challenge, we practice calculating the Pearson correlation coefficient.//
#include <cmath>
#include <cstdio>
#include <vector>
#include <iostream>
#include <algorithm>
#include <iomanip>
using namespace std;
float covariance(vector<float> x, vector<float> y, float meanX, float meanY){
float cov = 0.0f;
for(long unsigned i = 0; i < x.size(); i++){
cov += ((x[i] - meanX) * (y[i] - meanY));
}
return (cov / x.size());
}
float stdDev(vector<float> v, float mean){
float denominator = 0.0f;
for(long unsigned i = 0; i < v.size(); i++){
denominator += pow((v[i] - mean), 2);
}
return (sqrt(denominator / v.size()));
}
int main() {
/* Enter your code here. Read input from STDIN. Print output to STDOUT */
int n = 0;
vector<float> x;
vector<float> y;
float curr, meanX, meanY, sumX, sumY = 0.0f;
cin >> n;
for(int i = 0; i < n; i++){
cin >> curr;
x.push_back(curr);
sumX += curr;
}
for(int i = 0; i < n; i++){
cin >> curr;
y.push_back(curr);
sumY += curr;
}
meanX = sumX / x.size();
meanY = sumY / y.size();
float stdDevX = stdDev(x, meanX);
float stdDevY = stdDev(y, meanY);
float cov = covariance(x,y,meanX, meanY);
float pearsonCoef = cov / (stdDevX * stdDevY);
cout << fixed << setprecision(3) << pearsonCoef;
return 0;
}