Codeforces 1295E. Permutation Separation （线段树）

https://codeforces.com/contest/1295/problem/E

建一颗线段树，叶子结点是花费从1到i所需要花费的前缀和，表示前i个元素全部移动到右边的花费，再维护区间最小值，然后从1到n-1扫一遍，对于第i个位置，找到数字i在序列中的位置 pos ，将区间1到pos-1加上数字i移动的花费，pos到n-1减去数字i移动的花费，因为位置大于等于i 的时候，不需要将数字i移动到右边，位置小于i 时，需要把数字i移动到左边，所以需要增加数字i的花费，结合线段树维护的是前缀和数组，那么对于每一个位置i 用线段树维护的最小值去更新答案ans即可。

 #include<bits/stdc++.h>
 typedef long long ll;
 using namespace std;
 const int maxn = 2e5+;
 ll pos[maxn],cost[maxn],a[maxn],sum[maxn];
 struct segT{
     ll l,r;
     ll dat,lz;
 }t[maxn*];
 void build(ll p,ll l,ll r){
     t[p].l = l,t[p].r = r;
     if(l == r) { t[p].dat = sum[l] ,t[p].lz = ;return;}
     ll mid = (l+r)/;
     build(p*,l,mid);
     build(p*+,mid+,r);
     t[p].dat = min(t[p*].dat ,t[p*+].dat );
 }
 void upd(ll p,ll L,ll R,ll v){
     if(t[p].l >= L&&t[p].r <= R ) {t[p].dat += v,t[p].lz +=v;return;}
     if (t[p].lz && L!=R ) {
     t[p * ].dat += t[p].lz ,t[p*+].dat +=t[p].lz  ;
     t[p * ].lz  += t[p].lz ,t[p*+].lz += t[p].lz;
     t[p].lz = ;
     }
     int mid = (t[p].l + t[p].r )/;
     if(L<=mid) upd(p*,L,R,v);
     if(R>mid) upd(p*+,L,R,v);
     t[p].dat = min(t[p*].dat ,t[p*+].dat );
 }
 int query(ll p,ll l,ll r){
     if(l<=t[p].l && r>=t[p].r ) return t[p].dat ;
     int mid = (t[p].l + t[p].r )/;
     int val = 0x3f3f3f3f;
     if(l<=mid) val = min(val,query(p*,l,r));
     if(r>mid)  val = min(val,query(p*+,l,r));
     return val;
 }
 int main()
 {
     int n;
     scanf("%d",&n);
     for(int i = ;i<=n;i++) {
         scanf("%lld",&a[i]);
         pos[a[i]] = i;
     }
     for(int i = ;i<=n;i++){
         scanf("%lld",&cost[i]);
         sum[i] = sum[i-] + cost[i];
     }
     build(,,n-);
     ll ans = cost[n];
     ans = min(ans,t[].dat );
     for(int i = ;i<=n;i++){
         if(pos[i]!=n) upd(,pos[i],n-,-cost[pos[i]]);
         if(pos[i]!=) upd(,,pos[i]-,cost[pos[i]]);
         ans = min(ans,t[].dat);
     }
     printf("%lld",ans);
     return ;
 }