codeforces 819 B(区间加线性函数值)

问题描述

首先我们考虑这样一个问题。
给你一个操作

add [l,r] k(x-l)+b 对于 $x \in [l,r],d[x]+=k(x-r)+b$

由于这个不是常数，我们没法用线段树或者树状数组来做，但是我们可以想到，如果是常数，我们一定可以这样

$d[l]+=b,d[r+1]-=b;$
然后都操作完成之后$d[i] = \sum_{j=0}^id[j]$

这样就是$O(n)的$。
但是前面还有一个线性怎么办呢？

我们可以构造一个辅助数组 $df[]$ 记录每一段上的增量，(有点像微积分里的导数);

具体来说可以这样

$df[l+1]+=k,df[r+1]-=k$,
同理最后每个点上的增量就是 $df[i]= \sum_{j=0}^idf[i]$
而 $d[i] = d[i]+(df[i]= \sum_{j=0}^idf[i])$ ,最后再把 $d[i]$ 累加,
而需要注意的一点是由于有了增量操作, $d[r+1]$ 不应该只是加常数了，因为他要抵消前面的加和，这个可以简单计算一下 $d[r+1]-=b+k*(r-l)$

summary

最后我们就可以得出计算 $n$ 次操作后每个 $d[i]$ 的值了，具体代码如下

inline void add(LL k,LL b, int l,int r){
  if(l>r)return;
  d[l]+=b;
  d[r+1]-=b+k*(r-l);
  df[l+1]+=k;
  df[r+1]-=k;
}

void clc(){
  LL sum =0;
  FOR(0,n){
    sum+=df[i];
    d[i]+=sum;
  }
  FOR(1,n){
    d[i]+=d[i-1];
  }
}

参考链接

codeforces 819B题分析

codeforces B. Mister B and PR Shifts

题目见上面链接

设 $d_i$ 表示题意中 $k=i$时的计算值 那么 $p_k$ 对 $d_i$ 的贡献值为

$d_0+=|p_k-k|$
$d_1+=|p_k-(k+1)|$
$\dots$
$d_{n-k}+=|p_k-(n)|$
$d_{n-k+1}+=|p_k-1|$
$\dots$
$d_{n-1}+=|p_k-(k-1)|$

我们可以对 $p_k$ 的值做讨论，分$p_k\ge k \ or \ p_k\le k$,把家和分段,具体可参考我的代码或则上面的参考链接,

AC code

#include<bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define INF 0x3f3f3f3f
#define ALL(x) (x.begin(),x.end())
#define ms(x,v) memset(x,v,sizeof(x))
#define FORN(x,n) for(int i=x ; i<=n ; ++i)
#define FOR(x,n) for(int i=x ; i<n ; ++i)
using namespace std;
typedef long long LL;
typedef pair<LL,LL> Pair;

const int maxn = 1e6+10;
int n;
int p[maxn];//1;
LL d[maxn],df[maxn];

inline void add(LL k,LL b, int l,int r){
  if(l>r)return;
  d[l]+=b;
  d[r+1]-=b+k*(r-l);
  df[l+1]+=k;
  df[r+1]-=k;
}

void clc(){
  LL sum =0;
  FOR(0,n){
    sum+=df[i];
    d[i]+=sum;
  }
  FOR(1,n){
    d[i]+=d[i-1];
  }
}

int main(){

  ios::sync_with_stdio(false);
  cin.tie(0);
  ms(d,0);ms(df,0);
  cin>>n;
  FORN(1,n)cin>>p[i];

  FORN(1,n){
    if(p[i]>=i){
      add(-1,p[i]-i,0,p[i]-i);
      add(1,1,p[i]-i+1,n-i);
      add(-1,p[i]-1,n-i+1,n-1);
    }else{
      add(1,i-p[i],0,n-i);
      add(-1,p[i]-1,n-i+1,n-i+p[i]);
      add(1,1,n-i+p[i]+1,n-1);
    }

  }
  clc();
  int idx = min_element(d,d+n)-d;
  LL ans = d[idx];
  cout<<ans<<" "<<idx<<"\n";


  return 0;
  }