AtCoder ABC172 C問題でACできません

Question

C問題　Tundoku

ABC172のC問題でACできません。

queueで２つの本の山の先頭でより小さい方を選んで、残りの時間を順に減らしていこうとしたのですが、テストケースが大きいとREやWAが出てしまいます。

何回も調べてみてわからなかったのでよろしくお願い致します。

ソースコード

#include <iostream>
#include <queue>
using namespace std;

int solve (int &l ,int &k , queue<int> &a ,  queue<int> &b) {
  int ans = 0 , aa, bb;
  for (int i = 0 ; i < l ; i ++) {
    aa = a.front();
    bb = b.front();

    if (b.empty() || aa <= bb) { //Bが空かAの先頭がより小さいならAを選ぶ
      ans ++;
      k -= aa;
      a.pop();
      // cout << "a " << aa << endl;
    } else if (a.empty() || aa > bb) {
      ans ++;
      k -= bb;
      b.pop();
      // cout << "b " << bb << endl;
    }

    // cout << "k " << k << endl;
    // cout << "ans " << ans << endl;

    if (k == 0) {
      break;
    } else if (k < 0){ //Kが小さくなりすぎたら
      ans --;
      break;
    }

  }

  return ans;

}

int main(){
  int n , m , k , tmp;
  cin >> n >> m >> k;
  queue<int> a , b;
  for (int i = 0 ; i < n ; i ++) { cin >> tmp; a.push(tmp); }
  for (int i = 0 ; i < m ; i ++) { cin >> tmp; b.push(tmp); }
  int l = n + m;

  cout << solve(l, k , a , b) << endl;
}

Answer 1

公式の解説動画をご覧ください。すぬけさんがわかりやすく解説しておられます。
https://youtu.be/v8ppNGf49Nk?t=4795

PDFよりだいぶわかりやすいので毎回観ることをオススメします。
コンテスト終了直後からライブ配信されますが、リアルタイムで観ている場合はチャットで質問できます。

Comments

[Yug00]

ありがとうございます！
いつもPDFだけ見てました。
こっちも見てみようと思います！

Answer 2

机Aと机Bの現在の最上段のみを見て次に読むべき本を決めていることが問題です
例えば

5 5 150
10 30 10 10 10
20 20 20 20 20

のような入力が合ったとき、どうあるべきかを考えてみてください

Comments

[Yug00]

ありがとうございます！
より長い視点が要ったんですね。
何がいけないのか分かりました！