题目描述
给定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;
}