Submission #1162897

#TimeUsernameProblemLanguageResultExecution timeMemory
1162897Zero_OPMarathon Race 2 (JOI24_ho_t3)C++20
0 / 100
22 ms47684 KiB
#include <bits/stdc++.h>

using namespace std;

#define FOR(i, l, r) for(int i = (l); i < (r); ++i)
#define ROF(i, r, l) for(int i = (r)-1; i >= (l); --i)

#define mp make_pair
#define mt make_tuple
#define ff first
#define ss second

#define all(v) begin(v), end(v)
#define rall(v) rbegin(v), rend(v)
#define sz(v) (int)v.size()
#define pb push_back
#define eb emplace_back
#define sum_of(v) accumulate(all(v), 0ll)

#define dbg(x) "[" #x " = " << (x) << "]"

template<typename T>
      bool minimize(T& a, const T& b){
            if(a > b) return a = b, true;
            return false;
      }

template<typename T>
      bool maximize(T& a, const T& b){
            if(a < b) return a = b, true;
            return false;
      }

using ll = long long;
using db = double;
using ld = long double;
using ull = unsigned long long;

using pi = pair<int, int>;
using pl = pair<ll, ll>;
using pd = pair<db, db>;

using vi = vector<int>;
using vl = vector<ll>;
using vb = vector<bool>;
using vc = vector<char>;

using vpi = vector<pi>;
using vpl = vector<pl>;

void setIO(){
      ios_base::sync_with_stdio(0); cin.tie(0);
#ifdef LOCAL
      freopen("task.inp", "r", stdin);
      freopen("task.out", "w", stdout);
#endif // LOCAL
}

const int MAX = 5e5 + 5;
const int MAXDIFF = 1005;

int N, L, M, X[MAX], v[MAX], c[MAX], pref[MAX];
ll f[MAXDIFF][MAXDIFF][2], g[MAXDIFF][MAXDIFF][2];
//f[l][r][t] : [1...l] [r...N]] (l < r) if t = 0 : at L, t = 1 : at R, take the leftmost balls first
//f[l][r][t] : [1...l] [r...N]] (l < r) if t = 0 : at L, t = 1 : at R, take the rightmost balls first

ll dp_start_1[MAX][2]; //0 : v[i] > G, 1 : v[i] < G
ll dp_start_M[MAX][2]; //0 : v[i] > G, 1 : v[i] < G

int query_balls(int l, int r){
      if(l > r) return 0;
      assert(1 <= l && r <= M);
      assert((pref[r] - pref[l-1]) > 0);
      return pref[r] - pref[l-1];
}

int main(){
      setIO();

      cin >> N >> L;
      FOR(i, 0, N){
            cin >> X[i];
      }

      sort(X, X + N);
      FOR(i, 0, N){
            if(!M || v[M] != X[i]){
                  ++M;
                  v[M] = X[i];
                  c[M] = 1;
            } else ++c[M];
      }

      FOR(i, 1, M+2) pref[i] = pref[i-1] + c[i];

      int Q;
      cin >> Q;

      const int BOUND = 5e5;
      if(M > 1000){
            while(Q--){
                  int S, G, T;
                  cin >> S >> G >> T;
                  cout << "No\n";
            }
            return 0;
      }

      memset(f, 0x3f, sizeof(f));
      memset(g, 0x3f, sizeof(g));

      f[1][M+1][0] = 0;
      g[0][M][1] = 0;

      for(int len = M; len >= 1; --len){
            for(int l = 0, r = len; r <= M+1; ++l, ++r){
//                  cout << dbg(l) << dbg(r) << '\n';
                  if(l>0) minimize(f[l][r][0], f[l-1][r][0] + 1LL * (query_balls(1, l-1) + query_balls(r, M) + 1) * (v[l] - v[l-1]));
                  if(l>0) minimize(f[l][r][0], f[l-1][r][1] + 1LL * (query_balls(1, l-1) + query_balls(r, M) + 1) * (v[r] - v[l]));

                  if(r<M+1) minimize(f[l][r][1], f[l][r+1][0] + 1LL * (query_balls(1, l) + query_balls(r+1, M) + 1) * (v[r] - v[l]));
                  if(r<M+1) minimize(f[l][r][1], f[l][r+1][1] + 1LL * (query_balls(1, l) + query_balls(r+1, M) + 1) * (v[r] - v[r-1]));

                  if(l>0) minimize(g[l][r][0], g[l-1][r][0] + 1LL * (query_balls(1, l-1) + query_balls(r, M) + 1) * (v[l] - v[l-1]));
                  if(l>0) minimize(g[l][r][0], g[l-1][r][1] + 1LL * (query_balls(1, l-1) + query_balls(r, M) + 1) * (v[r] - v[l]));

                  if(r<M+1) minimize(g[l][r][1], g[l][r+1][0] + 1LL * (query_balls(1, l) + query_balls(r+1, M) + 1) * (v[r] - v[l]));
                  if(r<M+1) minimize(g[l][r][1], g[l][r+1][1] + 1LL * (query_balls(1, l) + query_balls(r+1, M) + 1) * (v[r] - v[r-1]));

//                  cout << dbg(f[l][r][0]) << dbg(f[l][r][1]) << dbg(g[l][r][0]) << dbg(g[l][r][1]) << '\n';

            }
      }

      memset(dp_start_1, 0x3f, sizeof(dp_start_1));
      memset(dp_start_M, 0x3f, sizeof(dp_start_M));

      FOR(i, 1, M+1){
            //find all state that start somewhere and end at v[i]
            //f[i-1][i][1] : start at v[1], end at v[i] -> abs(S - v[1]) + abs(G - v[i]) * (N+1) + f[i-1][i][1]
            //f[i][i+1][0] : start at v[1], end at v[i] -> abs(S - v[1]) + abs(G - v[i]) * (N+1) + f[i][i+1][0]

            //g[i-1][i][1] : start at v[M], end at v[i] -> abs(S - v[M]) + abs(G - v[i]) * (N+1) + g[i-1][i][1]
            //g[i][i+1][0] : start at v[M], end at v[i] -> abs(S - v[M]) + abs(G - v[i]) * (N+1) + g[i][i+1][0]

//            cout << f[i-1][i][1] << ' ' << f[i][i+1][0] << ' ';
//            cout << g[i-1][i][1] << ' ' << g[i][i+1][0] << '\n';

            minimize(dp_start_1[v[i]][0], f[i-1][i][1] + 1LL * v[i] * (N+1));
            minimize(dp_start_1[v[i]][1], f[i-1][i][1] - 1LL * v[i] * (N+1));
            minimize(dp_start_1[v[i]][0], f[i][i+1][0] + 1LL * v[i] * (N+1));
            minimize(dp_start_1[v[i]][1], f[i][i+1][0] - 1LL * v[i] * (N+1));

            minimize(dp_start_1[v[i]][0], g[i-1][i][1] + 1LL * v[i] * (N+1));
            minimize(dp_start_1[v[i]][1], g[i-1][i][1] - 1LL * v[i] * (N+1));
            minimize(dp_start_1[v[i]][0], g[i][i+1][0] + 1LL * v[i] * (N+1));
            minimize(dp_start_1[v[i]][1], g[i][i+1][0] - 1LL * v[i] * (N+1));
      }

      FOR(i, 1, BOUND+1) {
            minimize(dp_start_1[i][1], dp_start_1[i-1][1]);
            minimize(dp_start_M[i][1], dp_start_M[i-1][1]);
      }

      ROF(i, BOUND+1, 0){
            minimize(dp_start_1[i][0], dp_start_1[i+1][0]);
            minimize(dp_start_M[i][0], dp_start_M[i+1][0]);
      }

//      cout << f[2][3][0] << '\n';

      while(Q--){
            int S, G, T;
            cin >> S >> G >> T;

            //0 : v[i] > G, 1 : v[i] < G
            ll best = 1e18;
            FOR(i, 1, M+1){
//                  ll a = abs(S - v[1]) + abs(G - v[i]) * (N+1) + f[i-1][i][1];
//                  ll b = abs(S - v[1]) + abs(G - v[i]) * (N+1) + f[i][i+1][0];
//                  ll c = abs(S - v[M]) + abs(G - v[i]) * (N+1) + g[i-1][i][1];
//                  ll d = abs(S - v[M]) + abs(G - v[i]) * (N+1) + g[i][i+1][0];
//                  cout << dbg(a) << dbg(b) << dbg(c) << dbg(d) << '\n';
                  if(i>1) minimize(best, abs(S - v[1]) + abs(G - v[i]) * (N+1) + f[i-1][i][1]);
                  if(i<M) minimize(best, abs(S - v[1]) + abs(G - v[i]) * (N+1) + f[i][i+1][0]);
                  if(i>1) minimize(best, abs(S - v[M]) + abs(G - v[i]) * (N+1) + g[i-1][i][1]);
                  if(i<M) minimize(best, abs(S - v[M]) + abs(G - v[i]) * (N+1) + g[i][i+1][0]);
//                  cout << dbg(best) << '\n';
            }
            best += N;
//            cout << best << '\n';
            cout << (best <= T ? "Yes\n" : "No\n");
      }

      return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...