Kattis Putovanje 帮助 C++

Question

我正在研究 Kattis 问题 Putovanje ( https://open.kattis.com/problems/putovanje )。 我手动输入了样本输入，我的程序输出与给定的样本输出匹配。 但是出于某种原因，Kattis 接受了我的程序并在检查我的代码中途说“错误的答案”。 .

这是我为该程序运行的代码：

#include <iostream>
#include <vector>

using namespace std;

int main()
{
    int numFruits, numStomach, temp;
    vector<int> fruitWeight;

    cin >> numFruits >> numStomach;

    for (int i = 0; i < numFruits; i++)
    {
        cin >> temp;
        fruitWeight.push_back(temp);
    }

    int edibleFruits = 0;
    int fruitAmount = 0;
    int vecSize = fruitWeight.size();

   for (int i = 0; i< vecSize; i++)
   {
    if (edibleFruits < numStomach)
      {
        edibleFruits += fruitWeight[i];
        fruitAmount++;
      }
    if (edibleFruits > numStomach)
      {
        edibleFruits -= fruitWeight[i];
      }  
    }

 cout << fruitAmount << endl;
 return 0;
}

我不知道为什么 Kattis 不接受我的回答，非常感谢任何帮助！

谢谢！

样本输入 1

5 5
3 1 2 1 1

示例输出 1

4

样本输入 2

7 5
1 5 4 3 2 1 1

示例输出 2

3

Answer 1

让我们从这部分开始：

if (edibleFruits < numStomach) {
    edibleFruits += fruitWeight[i];
    fruitAmount++;
}
if (edibleFruits > numStomach) {
    edibleFruits -= fruitWeight[i];
}  

当米尔斯拉夫吃得过多时，你忘记制作fruitAmount-- 。 尽管我建议将其替换为：

if (edibleFruits + fruitWeight[i] <= numStomach) {
    edibleFruits += fruitWeight[i];
    fruitAmount++;
}

它更紧凑。

下一部分是：

 for (int i = 0; i< vecSize; i++)
   {
    if (edibleFruits < numStomach)
      {
        edibleFruits += fruitWeight[i];
        fruitAmount++;
      }
    if (edibleFruits > numStomach)
      {
        edibleFruits -= fruitWeight[i];
      }  
    }

你总是从第一个水果开始吃，这并不总是最佳的。 您可以从任何水果开始进食。 例如，输入是下一个：

3 100
100 50 1

使用您当前的代码，您将吃掉100水果并跳过所有下一步。 所以你的答案是： 1 ，正确的答案是2 。 因为你可以从第二个水果开始吃，吃50和1 。

解决它的最简单方法是有 2 个 for 循环，其中第一个循环将迭代 Milslav 将开始进食的起始索引。 下一个循环将迭代到最后。 总起始指数的最大水果食用量就是答案。 像这样：

for (int start_eat_from = 0; start_eat_from < N; ++start_eat_from) {
     for (int i = start_eat_from; i < N; ++i) {
     //Eat everything if Milslav can and check that he ate the maximum amount.
     }
}

它将起作用，因为将只有10^6次迭代并且它将在 1 秒内完成。

如果N大于它，则需要使用一些动态编程。

更新：

总而言之，代码将是这样的：

for (int start_from = 0; start_from < numFruits; ++start_from) {
        edibleFruits = 0;
        int tmpFruitAmount = 0;
        for (int i = start_from; i < numFruits; ++i) {
            if (edibleFruits + fruitWeight[i] <= numStomach) {
                edibleFruits += fruitWeight[i];
                tmpFruitAmount++;
            }
        }
        fruitAmount = max(fruitAmount, tmpFruitAmount);
    }