Problem p02623 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <bits/stdc++.h>
using ll = long long;
#define REP(i, n) for (ll i = 0; i < ll(n); i++)
#define FOR(i, a, b) for (ll i = a; i <= ll(b); i++)
#define ALL(x) x.begin(), x.end()

#define INF (ll)1e9 // 10^9:∞
#define LLINF (ll)1e12
#define MOD (ll)(1e9 + 7) // 10^9+7:合同式の法
#define PI 3.141592653589
#define PB push_back
#define F first
#define S second
#define __MAGIC__                                                              \\
  ios::sync_with_stdio(false);                                                 \\
  cin.tie(nullptr);

#define YESNO(T)                                                               \\
  if (T) {                                                                     \\
    cout << "YES" << endl;                                                     \\
  } else {                                                                     \\
    cout << "NO" << endl;                                                      \\
  }
#define yesno(T)                                                               \\
  if (T) {                                                                     \\
    cout << "yes" << endl;                                                     \\
  } else {                                                                     \\
    cout << "no" << endl;                                                      \\
  }
#define YesNo(T)                                                               \\
  if (T) {                                                                     \\
    cout << "Yes" << endl;                                                     \\
  } else {                                                                     \\
    cout << "No" << endl;                                                      \\
  }

#define Graph vector<vector<int>>
#define PII pair<int, int>
#define VI vector<int>
#define VVI vector<vector<int>>
#define VPII vector<pair<int, int>>

#define DDD fixed << setprecision(10)
#define endl "\\n"

using namespace std;
/*..................DEFINE GLOBAL VARIABLES...................*/

/*.....................DEFINE FUNCTIONS ......................*/

/*.........................kemkemG0...........................*/
signed main() {
  __MAGIC__

  ll N, M, K;

  cin >> N >> M >> K;
  VI A(N), B(M);
  REP(i, N) { cin >> A[i]; }
  REP(i, M) { cin >> B[i]; }

  ll ans = 0;
  ll n = 0;
  ll time = 0;
  ll a = 0;

  REP(i, N) {
    if (time + A[i] <= K) {
      time += A[i];
      n++;
      if (i == N - 1) {
        a = N - 1;
      }
    } else {
      a = i - 1;
      break;
    } // 最後に入れたAのインデックス 入れれなかったら-1
  }
  // cout<<a<<endl;
  ll j = 0;
  for (ll i = a; -1 <= i; --i) {
    // cout<<i<<endl;
    while (j < M && time + B[j] <= K) {
      time += B[j];
      n++;
      j++;
    }
    ans = max(ans, n);
    if (i >= 0) {
      time -= A[i];
      n--;
    }

    // if(i==-1){break;}
  }

  cout << ans << endl;

  return 0;
}


Error: TLE

Fixed submission:

// インクルード(アルファベット順)
#include <algorithm>     //sort,二分探索,など
#include <bitset>        //固定長bit集合
#include <chrono>        //実行時間計測
#include <cmath>         //pow,logなど
#include <complex>       //複素数
#include <deque>         //両端アクセスのキュー
#include <functional>    //sortのgreater
#include <iomanip>       //setprecision(浮動小数点の出力の誤差)
#include <iostream>      //入出力
#include <iterator>      //集合演算(積集合,和集合,差集合など)
#include <map>           //map(辞書)
#include <numeric>       //iota(整数列の生成),gcdとlcm,accumulate
#include <queue>         //キュー
#include <set>           //集合
#include <stack>         //スタック
#include <string>        //文字列
#include <unordered_map> //順序保持しないmap
#include <unordered_set> //順序保持しないset
#include <utility>       //pair
#include <vector>        //可変長配列

using namespace std;
typedef long long ll;

// マクロ
// forループ
// 引数は、(ループ内変数,動く範囲)か(ループ内変数,始めの数,終わりの数)、のどちらか
// Dがついてないものはループ変数は1ずつインクリメントされ、Dがついてるものはループ変数は1ずつデクリメントされる
#define REP(i, n) for (ll i = 0; i < ll(n); i++)
#define REPD(i, n) for (ll i = n - 1; i >= 0; i--)
#define FOR(i, a, b) for (ll i = a; i <= ll(b); i++)
#define FORD(i, a, b) for (ll i = a; i >= ll(b); i--)
// xにはvectorなどのコンテナ
#define ALL(x) x.begin(), x.end()
#define SIZE(x) ll(x.size())
// 定数
#define INF 1000000000000 // 10^12:∞
#define MOD 1000000007    // 10^9+7:合同式の法
#define MAXR 100000       // 10^5:配列の最大のrange
// 略記
#define PB push_back // 挿入
#define MP make_pair // pairのコンストラクタ
#define F first      // pairの一つ目の要素
#define S second     // pairの二つ目の要素

#define int long long

template <typename A, size_t N, typename T>

void Fill(A (&array)[N], const T &val) {
  std::fill((T *)array, (T *)(array + N), val);
}

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// ここにグローバル変数宣言
int H, W, K;
int I, J;
//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// ここに関数宣言

int calctime(int array[], int n) {
  int sum = 0;
  FOR(i, 0, n) { sum += array[i]; }
  return sum;
}

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
signed main() {
  // 入力の高速化用のコード
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  //////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

  int N, M, K;
  cin >> N >> M >> K;
  int A[N + 1], B[M + 1];
  A[0] = 0;
  B[0] = 0;
  FOR(i, 1, N) { cin >> A[i]; }
  FOR(i, 1, M) { cin >> B[i]; }

  int book = 0;

  int sum_b[M];

  FOR(i, 0, M) { sum_b[i] = calctime(B, i); }

  int sum = 0;

  REP(i, N + 1) {

    sum = calctime(A, i);

    if (sum > K) {
      break;
    }

    int nokori = 0;

    nokori = K - calctime(A, i);

    if (nokori < 0) {
      break;
    }

    // 2分探索用の変数(机Bに積んである本で2分探索)
    int left = 0;
    int right = M + 1;
    int mid = 0;

    while (right - left > 1) {

      // left と rightの中間地の計算
      mid = (left + right) / 2;

      // 中間地よりも大きかった場合は、leftに中間地の値を格納
      if (sum_b[mid] <= nokori)
        left = mid;
      // 中間地よりも小さかった場合は、rightに中間地の値を格納
      else
        right = mid;
    }

    book = max(book, i + left);
  }

  cout << book << endl;

  return 0;
}