线性相关系数，vectorNode完善拷贝

Author: sg-first

help.h

@@ -88,4 +88,12 @@ class help
         getchar();
     }
 
+    static vectorNode toVectorNode(vector<string>com)
+    {
+        vectorNode v(com.size()-1);
+        for(unsigned int i=1;i<com.size();i++)
+            v.v[i-1]=help::tofloat(com[i]);
+        return v;
+    }
+
 };

lab.h

@@ -1,28 +1,62 @@
 #pragma once
 #include "help.h"
 
-
 class lab
 {
-public:
-    static float avg(vector<string>com)
+private:
+    static float avg(vectorNode v)
     {
         float sum=0;
-        for(unsigned int i=1;i<com.size();i++)
-            sum+=help::tofloat(com[i]);
-        return sum/(com.size()-1);
+        for(unsigned int i=0;i<v.getl();i++)
+            sum+=v.v[i];
+        return sum/v.getl();
     }
 
-    static float dev(vector<string>com)
+    static float err(vectorNode v, float avg=-1)
     {
-        float avg=lab::avg(com);
+        if(avg==-1)
+        {
+            avg=lab::avg(v);
+        }
         float sum=0;
-        for(unsigned int i=1;i<com.size();i++)
+        for(unsigned int i=0;i<v.getl();i++)
         {
-            float num=help::tofloat(com[i])-avg;
+            float num=v.v[i]-avg;
             sum+=num*num;
         }
-        return sum/(com.size()-2);
+        return sum;
+    }
+
+public:
+    static float avg(vector<string>com)
+    {
+        //vectorNode v(com.size()-1);
+        vectorNode v=help::toVectorNode(com);
+        return avg(v);
+    }
+
+    static float var(vector<string>com)
+    {
+        auto v=help::toVectorNode(com);
+        return err(v)/v.getl();
+    }
+
+    static float r(vectorNode &x, vectorNode &y)
+    {
+        float avgx=avg(x);
+        float avgy=avg(y);
+        float lxx=err(x,avgx);
+        float lyy=err(y,avgy);
+
+        float lxy=0;
+        for(unsigned int i=0;i<x.getl();i++)
+        {
+            float num1=x.v[i]-avgx;
+            float num2=y.v[i]-avgy;
+            lxy+=abs(num1*num2);
+        }
+
+        return lxy/(sqrt(lxx*lyy));
     }
 
     static tuple<double,double> LeastSquare(vectorNode &x, vectorNode &y)

main.cpp

@@ -10,6 +10,14 @@
 
 using namespace std;
 
+void getXY(vectorNode &v1,vectorNode &v2)
+{
+    cout<<"vector x="<<endl;
+    help::getvector(v1);
+    cout<<"vector y="<<endl;
+    help::getvector(v2);
+}
+
 int main()
 {
     while(1)
@@ -61,26 +69,32 @@ int main()
                 help::t();
                 cout<<lab::avg(com)<<endl;
             }
-            if(com[0]=="dev")
+            if(com[0]=="var")
             {
                 isCommand=true;
                 help::t();
-                cout<<lab::dev(com)<<endl;
+                cout<<lab::var(com)<<endl;
             }
             if(com[0]=="linerFit")
             {
                 isCommand=true;
                 vectorNode v1(help::toint(com[1]));
                 vectorNode v2(help::toint(com[1]));
-                cout<<"vector x="<<endl;
-                help::getvector(v1);
-                cout<<"vector y="<<endl;
-                help::getvector(v2);
+                getXY(v1,v2);
                 double k,b;
                 tie(k,b)=lab::LeastSquare(v1,v2);
                 help::t();
                 cout<<"k:"<<k<<" b:"<<b<<endl;
             }
+            if(com[0]=="r")
+            {
+                isCommand=true;
+                vectorNode v1(help::toint(com[1]));
+                vectorNode v2(help::toint(com[1]));
+                getXY(v1,v2);
+                help::t();
+                cout<<"r:"<<lab::r(v1,v2)<<endl;
+            }
         }
         if(isCommand==false)
         {

scmath/matrix.hpp

@@ -7,6 +7,13 @@ class vectorNode : public BasicNode
 {
 private:
     unsigned int l;
+    void copy(const vectorNode& m2)
+    {
+        this->l=m2.l;
+        this->v = new double[l];
+        for (unsigned int i = 0; i < l; i++)
+            this->v[i] = m2.v[i];
+    }
 
 public:
     //SCMath兼容
@@ -27,10 +34,14 @@ class vectorNode : public BasicNode
 
     vectorNode(const vectorNode& m2) : BasicNode(m2)
     {
-        this->l = m2.l;
-        this->v = new double[l];
-        for (unsigned int i = 0; i < l; i++)
-            this->v[i] = m2.v[i];
+        copy(m2);
+    }
+
+    vectorNode& operator=(const vectorNode& m2)
+    {
+        delete []v;
+        copy(m2);
+        return *this;
     }
 
     double dot(vectorNode &v2)
@@ -76,7 +87,10 @@ class vectorNode : public BasicNode
         printf("\n");
     }
 
-    ~vectorNode() { delete[]v; }
+    ~vectorNode()
+    {
+        delete[]v;
+    }
     unsigned int getl() { return l; }
 };