Submission #1338211

#TimeUsernameProblemLanguageResultExecution timeMemory
1338211huutuanPeru (RMI20_peru)C++20
49 / 100
1102 ms200620 KiB
#include "peru.h"

#include <bits/stdc++.h>

using namespace std;

#define int long long

struct Node{
   int val, lazy;
   Node (int _val=1e18){
      val=_val, lazy=0;
   }
   Node operator+(Node a){
      return Node(min(val, a.val));
   }
};

struct SegmentTree{
   vector<Node> t;
   void init(int n){
      t.assign(4*n, Node());
   }
   void apply(int k, int val){
      t[k].val+=val;
      t[k].lazy+=val;
   }
   void push(int k){
      if (t[k].lazy){
         apply(k<<1, t[k].lazy);
         apply(k<<1|1, t[k].lazy);
         t[k].lazy=0;
      }
   }
   void update(int k, int l, int r, int L, int R, int val){
      if (r<L || R<l) return;
      if (L<=l && r<=R){
         apply(k, val);
         return;
      }
      push(k);
      int mid=(l+r)>>1;
      update(k<<1, l, mid, L, R, val);
      update(k<<1|1, mid+1, r, L, R, val);
      t[k]=t[k<<1]+t[k<<1|1];
   }
   void update_pos(int k, int l, int r, int pos, int val){
      if (l==r){
         t[k].val=val;
         return;
      }
      push(k);
      int mid=(l+r)>>1;
      if (pos<=mid) update_pos(k<<1, l, mid, pos, val);
      else update_pos(k<<1|1, mid+1, r, pos, val);
      t[k]=t[k<<1]+t[k<<1|1];
   }
   Node get(int k, int l, int r, int L, int R){
      if (r<L || R<l) return t[0];
      if (L<=l && r<=R) return t[k];
      push(k);
      int mid=(l+r)>>1;
      return get(k<<1, l, mid, L, R)+get(k<<1|1, mid+1, r, L, R);
   }
} seg;

const int N=2.5e6+10;
int f[N], a[N];

const int mod=1e9+7;

int32_t solve(int32_t n, int32_t k, int32_t *aa){
   seg.init(n);
   stack<int> st;
   seg.update_pos(1, 1, n, 1, 0);
   st.push(0);
   a[0]=0;
   for (int i=1; i<=n; ++i) a[i]=aa[i-1];
   int ans=0;
   for (int i=1; i<=n; ++i){
      while (st.size() && a[st.top()]<a[i]){
         int x=a[st.top()];
         int r=st.top();
         st.pop();
         int l=st.size()?st.top()+1:1;
         seg.update(1, 1, n, l, r, a[i]-x);
      }
      seg.update(1, 1, n, i, i, a[i]);
      st.push(i);
      if (i==1) f[i]=a[1];
      else f[i]=seg.get(1, 1, n, max((int)1, i-k+1), i).val;
      if (i<n) seg.update_pos(1, 1, n, i+1, f[i]);
      ans=(ans*23+f[i])%mod;
   }
   return ans;
}

#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...