2013-2014 ACM-ICPC, NEERC, Southern Subregional Contest Problem L. Stock Trading Robot 水题

Problem L. Stock Trading Robot

题目连接：

http://www.codeforces.com/gym/100253

Description

CyberTrader is an all-in-one trading solution for investment banks, quantitative hedge funds and

proprietary trading groups. It has only one drawback it is not implemented yet.

You are working on implementing a simple algorithm to buy/sell shares. It should work as follows. Initially

a robot has d dollars and doesn't have any shares. The robot's behaviour is dened by two positive integer

numbers a and b, their role is explained below.

Starting from the second day, every day the robot analyzes a new share price comparing it with the

previous share price. If the price increases the robot buys shares it buys as many shares as it can but

not more than x. Actually, x is not a constant and depends on the number of consecutive increases: x = a

for the rst increase, x = 2a for two increases in a row, and so on, i.e. x = ka for k consecutive increases.

Surely, the robot can buy only non-negative integer number of shares and the number depends on the

money it has and on x.

If the price decreases the robot sells shares it sells as many shares as it has but not more than y.

Actually, y is not a constant and depends on the number of consecutive decreases: y = b for the rst

decrease in a row, y = 2b for two decreases in a row, and so on, i.e. y = kb for k consecutive decreases.

If the price doesn't change the robot does not buy or sell any shares.

Write a program for the robot to simulate the above algorithm.

Input

The rst line of the input contains four positive integers n, d, a and b (1 ≤ n, d ≤ 105

, 1 ≤ a, b ≤ 10),

where n is the number of days to simulate the algorithm. The following line contains sequence of positive

integers p1, p2, . . . , pn (1 ≤ pi ≤ 105

), where pi

is the share price on the i-th day.

It is guaranteed that there will be no overow of the 32-bit signed integer type, so feel free to use type

int (in C++ or Java) to store the number of dollars and shares.

Output

Print the number of dollars and the number of shares the robot will have after n days.

Sample Input

5 10 1 2

1 2 3 4 5

Sample Output

2 3

Hint

题意

自动炒股机器人，在什么价格的时候，会自动买，然后自动卖。

题面给你说买的条件，和卖的条件。

题解：

模拟就好了……

代码

#include <bits/stdc++.h>

using namespace std;

const int N=100010;
int n,a,b,d,p[N];

int main()
{
    scanf("%d%d%d%d",&n,&d,&a,&b);
    for(int i=0;i<n;i++)
        scanf("%d",&p[i]);
    int cnt1=0,cnt2=0;
    long long ans1=1LL*d,ans2=0;
    for(int i=1;i<n;i++)
    {
        if(p[i]>p[i-1])
        {
            cnt1++;cnt2=0;
            long long tmp=min(1LL*cnt1*a,ans1/p[i]);
            ans1-=tmp*p[i];
            ans2+=tmp;
        }
        else
        if(p[i]<p[i-1])
        {
            cnt2++;cnt1=0;
            long long tmp=min(1LL*cnt2*b,ans2);
            ans1+=tmp*p[i];
            ans2-=tmp;
        }
        else cnt1=0,cnt2=0;
    }
    cout<<ans1<<" "<<ans2<<endl;
    return 0;
}