蓝莓 (SPOJ) - 超出动态编程时间限制

Blueberries (SPOJ) - Dynamic Programming Time Limit Exceeded

本文关键字:编程 动态 时间 SPOJ 蓝莓      更新时间:2023-10-16

我正在解决一个关于spoj的问题。问题有一个简单的递归解决方案。

问题:给定一个大小为 n 的数字数组,选择一组数字,以便集合中没有两个元素是连续的,并且子集元素的总和将尽可能接近 k,但不应超过它。

我的递归方法

我使用了一种类似于背包的方法,将问题划分为一个包含当前元素而另一个忽略它。

  function solve_recursively(n, current, k)
     if n < 0
        return current
     if n == 0
        if current + a[n] <= k
           return current + a[n]
        else
           return current
     if current + a[n] > k
        return recurse(n-1, current, k)
     else
        return max(recurse(n-1, current, k), recurse(n-2, current+a[n], k))

后来由于它本质上是指数级的,我使用 map(在 C++ 年)来做记忆以降低复杂性。

我的源代码:

struct k{
  int n; 
  int curr;
};
bool operator < (const struct k& lhs, const struct k& rhs){
  if(lhs.n != rhs.n)
    return lhs.n < rhs.n;
  return lhs.curr < rhs.curr;
};
int a[1001];
map<struct k,int> dp;
int recurse(int n, int k, int curr){
  if(n < 0)
    return curr;
  struct k key = {n, curr};
  if(n == 0)
    return curr + a[0] <= k ? curr + a[0] : curr;
  else if(dp.count(key))
    return dp[key];
  else if(curr + a[n] > k){
    dp[key] = recurse(n-1, k, curr);
    return dp[key];
  }
  else{
    dp[key] = max(recurse(n-1, k, curr), recurse(n-2, k, curr+a[n]));
    return dp[key];
  }
}
int main(){
  int t,n,k;
  scanint(t);
  while(t--){
    scanint(n);
    scanint(k);
    for(int i = 0; i<n; ++i)
      scanint(a[i]);
    dp.clear();
    printf("Scenario #%d: %dn",j, recurse(n-1, k, 0));
  }
  return 0;
}

我检查了给定的测试用例。它清除了他们。但是我在提交时得到了错误的答案。

编辑:早些时候我的输出格式是错误的,所以我得到了错误的答案。但是,现在它显示已超过时间限制。我认为自下而上的方法会有所帮助,但我在制定方法时遇到了问题。我把它当作自下而上的背包,但在确切的配方上遇到了一些困难。

据我了解,您几乎有解决方案。如果递归关系正确但效率太低,您只需将递归更改为迭代即可。显然,您已经有了表示状态及其各自值的数组dp。基本上,你应该能够用三个嵌套循环来解决填充dp,分别用于nkcurr,这将分别增加以确保所需的每个值都已经计算dp。然后,您将对 recurse 的递归调用替换为对 dp 的访问。