제출 #1136085

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1136085		mychecksedad	Mizuyokan 2 (JOI23_mizuyokan2)	C++20	0 / 100	4094 ms	5296 KiB

mizuyokan2

/* Author : Mychecksdead  */
#include<bits/stdc++.h>
using namespace std;
#define ll long long int
#define MOD (1000000000+7)
#define MOD1 (998244353)
#define pb push_back
#define all(x) x.begin(), x.end()
#define en cout << '\n'
#define ff first
#define ss second
#define pii pair<int,int>
#define vi vector<int>
const int N = 2e5+100, M = 1e5+10, K = 31, MX = 200;


int n, ANS[N], GO_SMALL[N], GO_BIG[N];
ll a[N], pref[N];

void solv(int A, int B, array<int, 2> *dp){
  dp[0][0] = dp[0][1] = 0;
  dp[1][0] = 1;
  dp[1][1] = 1;
  for(int j = A + 1; j <= B; ++j){
    // int l = A, r = j - 1, go = j;
    // while(l <= r){
    //   int m = l + r>>1;
    //   if(pref[j - 1] - pref[m - 1] > a[j]){
    //     go = m;
    //     l = m + 1;
    //   }else r = m - 1;
    // }
    dp[j - (A - 1)][0] = dp[j - (A - 1)][1] = 1;
    if(GO_SMALL[j] != -1 && GO_SMALL[j] - (A - 1) >= 0){
      // for(int l = go; l >= max(A, go - K); --l){
      //   if(l == A || pref[j - 1] - pref[l - 1] > a[l - 1]){
      dp[j - (A - 1)][0] = max(dp[j - (A - 1)][0], dp[GO_SMALL[j] - (A - 1)][0] + 2);
        // }
      // }
    }
    if(GO_BIG[j] != -1 && GO_BIG[j] - (A - 1) >= 0){
      dp[j - (A - 1)][1] = max(dp[j - (A - 1)][1], dp[GO_BIG[j] - (A - 1)][0] + 1);
    }
  }
}

void solv_rev(int A, int B, array<int, 2> *dp){
  // vector<array<int, 2>> dp(B - A + 2);
  reverse(a+A, a+B+1);
  pref[A] = pref[A - 1] + a[A];
  for(int j = A + 1; j <= B; ++j) pref[j] = pref[j - 1] + a[j];
  dp[0][0] = dp[0][1] = 0;
  dp[1][0] = 1;
  dp[1][1] = 1;
  for(int j = A + 1; j <= B; ++j){
      // dp[j][1] = dp[j - 1][1];
    int l = A, r = j - 1, go = j;
    while(l <= r){
      int m = l + r>>1;
      if(pref[j - 1] - pref[m - 1] > a[j]){
        go = m;
        l = m + 1;
      }else r = m - 1;
    }
    dp[j - (A - 1)][0] = dp[j - (A - 1)][1] = 1;
    if(go < j){
      for(int l = go; l >= max(A, go - K); --l){
        if(l == A || pref[j - 1] - pref[l - 1] > a[l - 1]){
          dp[j - (A - 1)][0] = max(dp[j - (A - 1)][0], dp[l - 1 - (A - 1)][0] + 2);
        }
      }
    }

    
  }
  reverse(a+A, a+B+1);
  pref[A] = pref[A - 1] + a[A];
  for(int j = A + 1; j <= B; ++j) pref[j] = pref[j - 1] + a[j];
}

array<int, 2> dpL[K + 5][N], dpR[K + 5][N];
array<int, 2> go_left[N], go_right[N];

void dfs(int l, int r, vector<array<int, 3>> &Q){
  vector<array<int, 3>> L, R, CUR;
  int m = l+r>>1;
  for(auto [x, y, idx]: Q){
    if(y <= m) L.pb({x, y, idx});
    else if(x >= m + 1) R.pb({x, y, idx});
    else CUR.pb({x, y, idx});
  }
  if(CUR.size()){
    if(l == r){
      for(auto [x, y, idx]: CUR){
        ANS[idx] = 1;
      }
      return;
    }
    // vector<vector<array<int, 2>>> dpL;
    int c = 0;
    for(int j = m; j >= max(l, m - K); --j){
      ++c;
      solv_rev(l, j, dpL[c - 1]);
    }
    c = 0;
    for(int j = m + 1; j <= min(r, m + 1 + K); ++j){
      ++c;
      solv(j, r, dpR[c - 1]);
    }
    solv(m, r, go_right);
    solv_rev(l, m + 1, go_left);
    for(auto [x, y, idx]: CUR){
      int ans = 1;
      for(int left = m; left >= max(x, m - K + 1); --left){
        for(int right = m + 1; right <= min(y, m + K); ++right){
          bool aight = true;
          if(!(left == x || pref[right] - pref[left - 1] > a[left - 1])) aight = false;
          if(!(right == y || pref[right] - pref[left - 1] > a[right + 1])) aight = false;
          if(aight){
            int cost = 1 + (left == x ? 0 : max(dpL[m-(left-1)][(left-1)-(x)+1][0], dpL[m-(left-1)][(left-1)-(x)+1][1])) 
              + (right == y ? 0 : max(dpR[(right+1)-(m+1)][(y)-(right+1)+1][0], dpR[(right+1)-(m+1)][(y)-(right+1)+1][1]));
            ans = max(ans, cost);
          }
        }
      }
      ans = max({ans,
        max(go_right[y - m + 1][0], go_right[y - m + 1][1]) + max(dpL[0][m - x + 1][0], dpL[0][m - x + 1][1]) - 1, 
        max(go_left[m + 1 - x + 1][0], go_left[m + 1 - x + 1][1]) + max(dpR[0][y - (m + 1) + 1][0], dpR[0][y - (m + 1) + 1][1]) - 1, 
      });
      ANS[idx] = ans;
    }
  }


  if(l == r) return;
  dfs(l, m, L);
  dfs(m+1, r, R);
}
array<int, 2> dpp[N];
void solve(){
  cin >> n;
  for(int i = 1; i <= n; ++i){
    cin >> a[i];
  }
  pref[0] = 0;
  for(int i = 1; i <= n; ++i){
    pref[i] = pref[i - 1] + a[i];
  }

  for(int i = 1; i <= n; ++i){
    int l = 1, r = i - 1, go = i;
    while(l <= r){
      int m = l + r>>1;
      if(pref[i - 1] - pref[m - 1] > a[i]){
        go = m;
        l = m + 1;
      }else r = m - 1;
    }
    if(go < i){
      for(int l = go; l >= max(1, go - K); --l){
        if(l == 1 || pref[i - 1] - pref[l - 1] > a[l - 1]){
          GO_SMALL[i] = l - 1;
          break;
        }
      }
    }else GO_SMALL[i] = -1;
    GO_BIG[i] = -1;
    for(int l = i - 1; l >= max(1, (i-K)); --l){
      if(pref[i] - pref[l] > a[l]){
        GO_BIG[i] = l;
        break;
      }
    }
  }

  vector<array<int, 3>> Q;
  int q; cin >> q;
  for(int rep = 0; rep < q; ++rep){
    int x, y, A, B; cin >> x >> y >> A >> B;
    Q.pb({A+1, B, rep});
    ++A;
    a[x] = y;
    pref[0] = 0;
    for(int j = 1; j <= n; ++j){
      pref[j] = pref[j - 1] + a[j];
    }
    for(int i = 1; i <= n; ++i){
      int l = 1, r = i - 1, go = i;
      while(l <= r){
        int m = l + r>>1;
        if(pref[i - 1] - pref[m - 1] > a[i]){
          go = m;
          l = m + 1;
        }else r = m - 1;
      }
      GO_SMALL[i] = -1;
      if(go < i){
        for(int l = go; l >= max(1, go - K); --l){
          if(l == 1 || pref[i - 1] - pref[l - 1] > a[l - 1]){
            GO_SMALL[i] = l - 1;
            break;
          }
        }
      }else GO_SMALL[i] = -1;
      GO_BIG[i] = -1;
      for(int l = i - 1; l >= max(1, (i-K)); --l){
        if(pref[i] - pref[l] > a[l]){
          GO_BIG[i] = l;
          break;
        }
      }
    }
    // for(int i = 1; i <= n; ++i){
    //   cout << a[i] << ' ';
    // }
    // en;

    // for(int i = 1; i <= n; ++i){
    //   cout << GO_SMALL[i] << ' ';
    // }
    // en;
    // for(int i = 1; i <= n; ++i){
    //   cout << GO_BIG[i] << ' ';
    // }
    // en;
    solv(A, B, dpp);
    cout << max(dpp[B-A+1][0], dpp[B-A+1][1]) << '\n';
  }
  // dfs(1, n, Q);


  // for(int i = 1; i <= q; ++i){
  //   cout << ANS[i-1] << '\n';
  // }
}


int main(){
  cin.tie(0); ios::sync_with_stdio(0);
  int tt = 1, aa;
  // freopen("in.txt", "r", stdin);
  // freopen("out.txt", "w", stdout);
  while(tt--){
    solve();
    en;
  }
  cerr<<"time taken : "<<(float)clock()/CLOCKS_PER_SEC<<" seconds\n";
  return 0;
} 

• Subtask 1: 0 / 6
• Subtask 2: 0 / 9
• Subtask 3: 0 / 13
• Subtask 4: 0 / 32
• Subtask 5: 0 / 29
• Subtask 6: 0 / 11