2-5 E. 计算2支股票的M天运动平均价格

 题目描述

给定2支股票的开盘价和收盘价的N天历史数据，

要求按开盘和收盘，分别计算每支股票的每个日期对应的M天移动平均价格。

假定两个股票数据如下：

日期            开盘/收盘      第1支股票价格S1       第2支股票价格S2

2004/7/29 close 6 4

2004/7/25 close 2 6

2004/7/26 open 8 12

2004/7/30 open 2 4

2004/7/27 close 8 10

2004/7/28 open 4 2

按M=2天（日期不用连续）计算移动平均价格，按先开盘，后收盘价，输出如下：（若某日期之前，没有M-1条的记录（日期不用连续），则不用输出）

2004/7/28 open 6 7

2004/7/30 open 3 3

2004/7/27 close 5 8

2004/7/29 close 7 7

其中, 2004/7/28日的S1的值为(8+4)/2 = 6， 即将2004/7/28和（最近1条记录2004/7/26，最近2条记录，最近M-1条记录）的价格，求和并计算平均。

输入 

第1行：N天记录   M天平均

第2行到N+1行：N天2支股票的开盘与收盘价格（注意日期是无序的）

6  2

2004/7/29 close 6 4

2004/7/25 close 2 6

2004/7/26 open 8 12

2004/7/30 open 2 4

2004/7/27 close 8 10

2004/7/28 open 4 2

输出 

每个日期的最近M条记录（包括该日期的价格在内）的平均价格（若某日期之前没有M-1条的记录（日期不用连续），则不用输出）

2004/7/28 open 6 7

2004/7/30 open 3 3

2004/7/27 close 5 8

2004/7/29 close 7 7

代码 

#include <iostream>
using namespace std;

struct Stock{  //股票数据结构体
    string date;
    string situ;
    int first;
    int second;
};

class comb{
public:
    Stock *sto;
    int N,M;
    comb(){
        cin >> N >> M;
        sto = new Stock[N];
    }

    void create(){  //录入数据
        for(int i = 0; i < N; i++){
            cin >> sto[i].date >> sto[i].situ >> sto[i].first >> sto[i].second;
        }
    }

    void toSort(){  //按照open在前close在后以及日期先后冒泡排序
        Stock temp;
        for(int i = 1; i < N; i++){
            for(int j = 0; j < N-i; j++){
                if(sto[j].situ == "close" && sto[j+1].situ == "open"){
                    temp = sto[j];
                    sto[j] = sto[j+1];
                    sto[j+1] = temp;
                }else if(sto[j].situ == sto[j+1].situ){
                    if(sto[j].date > sto[j+1].date){
                        temp = sto[j];
                        sto[j] = sto[j+1];
                        sto[j+1] = temp;
                    }
                }
            }
        }
    }

    void toComb(){  //计算输出
        int flag1=0, flag2=0;
        int key1,key2;  //储存计算结果
        for(int i = 0; i < N; i++){  //对于该序列open与close位于第一个的sto都不计算输出
            if(flag1 == 1 && sto[i].situ != "close"){  //输出open开盘的数据
                key1 = (sto[i-1].first + sto[i].first)/M;  //不可直接改变sto的first、second值，因为后续日期计算依赖原数值
                key2 = (sto[i-1].second + sto[i].second)/M;
                cout << sto[i].date << " " << sto[i].situ << " " << key1 << " " << key2 << endl;
            }
            if(flag2 == 1){  //输出close收盘的数据
                key1 = (sto[i-1].first + sto[i].first)/M;
                key2 = (sto[i-1].second + sto[i].second)/M;
                cout << sto[i].date << " " << sto[i].situ << " " << key1 << " " << key2 << endl;
            }
            if(sto[i].situ == "open"){
                flag1 = 1;
                flag2 = 0;
            }else{
                flag2 = 1;
                flag1 = 0;
            }
        }
    }
};

int main()
{
    comb st;
    st.create();
    st.toSort();
    st.toComb();
    return 0;
}