7-1 大師と仙人との奇遇 (20 分)

7-1 大師と仙人との奇遇 (20 分)

刘大师是公认的JAVA达人，但对股票交易这门“技术”却是素人，每次都把握不准买卖时机，经常亏损。一天，大师郁闷地在街上溜达，满脑子都在思索如何预测股票走势的超难算法。这时，地面突然卷起一股风，不知从哪儿刮来一位相貌奇特的“仙人”。只见仙人递给大师一本封面大写着《如来神股》的书，微笑着说：“年轻人，我看你气色不佳，送你本祖传秘诀提提神吧”，话音刚落，仙人又随风飘走了。刘大师一脸懵逼地翻开书，发现里面只有八个字：细水长流、见好就收。“遇到鬼了?!”，刘大师嘀咕道，但把这两句话反复瞟了几眼后，突然茅舍顿开: “哇撒，原来简单的才是最好的!”。刘大师醒悟过来后，立马跑回家用不到三分钟的时间就编了一个炒股"神器"。现在请按大师的思路编程，验证一下该方法是否有效。
刘大师的神器有两个功能：第一，设置N天为一个周期，除最后一天不买之外，其它天每天都买进且只买一支股票，做到细水长流；第二，每天买进股票的同时，把之前买进的且还留在手头上的股票，只要买进价格比当天交易额低，就立马卖出赚差价，见好就收。如果到最后一天还有没卖掉的，就全部出售。

输入格式:

输入两行数据，第一行只有一个整数N (1<N≤10⁶ )。第二行有N个正整数，第i∈[1,N]个数s ,i ∈[1,10 ⁵ ]表示第i天的单支股票的交易额。数据间用空格分开，末尾的数后面没有空格。

输出格式:

输出一个整数，表示N天的股票交易总收益。

输入样例:

7
8 1 4 6 7 10 5

输出样例:

6

总支出：8+1+4+6+7+10 = 36
总收入：4+6+7+10+10+5 = 42
注：比如第一天买进的在第六天卖出，第二天买进的在第三天卖出，。。。以此类推。

题解

1

用vector(数组)每次循环时进行一次判断
时间复杂度O(n²)

#include <bits/stdc++.h>
using namespace std;

int main(){
    int n;
    long long pay,pri,sold=0;
    scanf("%d", &n);
    scanf("%lld", &pri);
    vector<int> num;
    num.push_back(pri);
    pay=pri;
    for(int i=1;i<n-1;i++){
        scanf("%lld", &pri);
        pay+=pri;
        for(int j=num.size()-1;j>=0;j--){
            if(pri>num[j]){
                num.erase(num.begin()+j,num.begin()+j+1);
                sold+=pri;
            }
        }
        num.push_back(pri);
    }
    scanf("%lld", &pri);
    sold+=pri*num.size();
    printf("%lld", sold-pay);
    // cout<<sold-pay;
}

num也可以用字典等方式实现

2

用优先队列（或栈）
时间复杂度为O(nlogn)

#include <bits/stdc++.h>
using namespace std;

int main(){
    int n;
    long long pay,pri,sold=0;
    scanf("%d", &n);
    scanf("%lld", &pri);
    priority_queue<int,vector<int>,greater<int> > num;
    // stack<int> num;
    num.push(pri);
    pay=pri;
    for(int i=1;i<n-1;i++){
        scanf("%lld", &pri);
        pay+=pri;
        num.push(pri);
        while(pri>num.top()){
            num.pop();
            sold+=pri;
        }
        
    }
    scanf("%lld", &pri);
    sold+=pri*num.size();
    printf("%lld", sold-pay);
    // cout<<sold-pay;
}