c++11 - 股票最大化问题的基于 DP 的算法 Hackerrank

Question

我需要一些帮助来解决hackerrank上的股票最大化问题问题在这个链接https://www.hackerrank.com/challenges/stockmax/problem 我知道有一个线性方法可以解决这个问题，但我需要动态编程方面的帮助我的一个解决方案。当我提交它时，它会给出 TLE。作为一名学习者，我只想知道是否可以对基于 DP 的解决方案进行进一步优化。我编写了以下函数来评估答案

long myans(vector<int> prices, unordered_map<int,unordered_map<int, long>> &mp ,int st,long ns,long pr)
{
   //base
   if(st==prices.size())
   return pr;
   //dp
   if( mp[ns][st]!=0)
   return  mp[ns][st];
   //rec
   long dn=0,ss=0,bs=0;
   if(st!=prices.size()-1)
   bs=myans(prices,mp,st+1,ns+1,pr-prices[st]);
   if(ns>0)
   ss=myans(prices,mp,st+1,0,pr+ns*prices[st]);
   dn=myans(prices,mp,st+1,ns,pr);
   mp[ns][st]=max(bs,max(ss,dn));
   // mp[st+1][ns+1]=bs;
   // mp[st+1][ns]=dn;
   // mp[st+1][0]=ss;
    return mp[ns][st];
   //return  mp[make_pair(ns,prices[st])];
}
long stockmax(vector<int> prices) {
unordered_map<int,unordered_map<int, long>> mp;
   return myans(prices,mp,0,0,0);

}

标签： c++11dynamic-programming