《数据科学面试40+真题讲解》,K神本年度最后一次开课


一亩三分地论坛

 找回密码
 Sign Up 注册获取更多干货
码农求职神器Triplebyte:
不用海投,内推你去多家公司面试
Airbnb 数据科学职位
in analytics and inference
天天打游戏、照样领工资,
你要不要来?
把贵司招聘信息放这里
查看: 849|回复: 4
收起左侧

[算法题] stock maximize 不能 pass tests, 求正确 java code

[复制链接] |试试Instant~ |关注本帖
yi1san3fendi 发表于 2016-2-21 07:50:58 | 显示全部楼层 |阅读模式

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看,没有帐号?Sign Up 注册获取更多干货

x
题目如下,可以直接google "hackrank  stock maximize",测试自己代码是否正确:
***hackerrank.com/challenges/stockmax  (*** 用 www.替代)

Your algorithms have become so good at predicting the market that you now know what the share price of Wooden Orange Toothpicks Inc. (WOT) will be for the next N days.

Each day, you can either buy one share of WOT, sell any number of shares of WOT that you own, or not make any transaction at all. What is the maximum profit you can obtain with an optimum trading strategy?

Input
The first line contains the number of test cases T. T test cases follow:
The first line of each test case contains a number N. The next line contains N integers, denoting the predicted price of WOT shares for the next N days.

Output
Output T lines, containing the maximum profit which can be obtained for the corresponding test case.

Constraints
1 <= T <= 10
1 <= N <= 50000
All share prices are between 1 and 100000

Sample Input
3
3
5 3 2
3
1 2 100
4
1 3 1 2

Sample Output
0
197
3

我按照网上方法先右遍历,再左遍历,java代码如下,但不能pass tests; 求正确java 代码

import java.io.*;
import java.util.*;
import java.text.*;
import java.math.*;
import java.util.regex.*;

public class Solution {

    public static void main(String[] args) {
        /* Enter your code here. Read input from STDIN. Print output to STDOUT. Your class should be named Solution. */
        Scanner sc = new Scanner(System.in);
       int numOfTestCase = sc.nextInt();
       for (int i = 0; i < numOfTestCase; i++) {
              int n = sc.nextInt();
              int[] stockPrice = new int[n];

              for (int j = 0; j < n; j++) {
                    stockPrice[j] = sc.nextInt();
              }
            
              int max = 0;
              int[] leftMax = new int[n];
              for (int j = n - 1; j >= 0; j--) {
                  if(stockPrice[j] > max)
                      max = stockPrice[i];
                  leftMax[j] = max;
              }
              long profit = 0;
              for(int k=0;k<n; k++){
                  if(leftMax[k] > stockPrice[k])
                      profit += leftMax[k]-stockPrice[k];
              }
              System.out.println(profit);
       }
       sc.close();
    }
}


stellari 发表于 2016-2-21 14:03:17 | 显示全部楼层
这两句看出点什么问题没有?
  1.                   
  2. if(stockPrice[j] > max)
  3.    max = stockPrice[i];
复制代码
另外,这题不需要从右往左再从左往右,只要在从右往左扫时顺便获得“第 i 天购买的1股能够获得的最大利润”即可。
回复 支持 1 反对 0

使用道具 举报

ctzsm 发表于 2016-2-21 13:26:42 | 显示全部楼层
本帖最后由 ctzsm 于 2016-2-21 13:30 编辑
  1. import java.io.*;
  2. import java.util.*;

  3. public class Solution {

  4.     public static void main(String[] args) {
  5.         Scanner in = new Scanner(new BufferedInputStream(System.in));
  6.         int T = in.nextInt();
  7.         while (T-- > 0) {
  8.             int n = in.nextInt();
  9.             int[] a = new int[50005];
  10.             for (int i = 0; i < n; ++i) {
  11.                 a[i] = in.nextInt();
  12.             }
  13.             int now = a[n - 1];
  14.             long ans = 0;
  15.             for (int i = n - 2; i >= 0; --i) {
  16.                 now = Math.max(now, a[i]);
  17.                 ans += now - a[i];
  18.             }
  19.             System.out.println(ans);
  20.         }
  21.     }
  22. }
复制代码

评分

1

查看全部评分

回复 支持 1 反对 0

使用道具 举报

 楼主| yi1san3fendi 发表于 2016-2-21 13:54:03 | 显示全部楼层

感谢你及时的回复!hackrank 2 cases 测试是对的;给你评了大米3升
回复 支持 反对

使用道具 举报

 楼主| yi1san3fendi 发表于 2016-2-21 14:55:05 | 显示全部楼层
stellari 发表于 2016-2-21 14:03
这两句看出点什么问题没有?另外,这题不需要从右往左再从左往右,只要在从右往左扫时顺便获得“第 i 天购 ...

是的,你说的没错
回复 支持 反对

使用道具 举报

本版积分规则

关闭

一亩三分地推荐上一条 /5 下一条

手机版|小黑屋|一亩三分地论坛声明

custom counter

GMT+8, 2017-11-24 19:34

Powered by Discuz! X3

© 2001-2013 Comsenz Inc. Design By HUXTeam

快速回复 返回顶部 返回列表