CF 820D Mister B and PR Shifts 差分(区间+k*(x-L)+b)

题意:给出[1..n]排列p 定义偏差值为segma(p[i]-i) i=1..n
操作:将每个元素向后移一位,最后一个元素移动到开头.
n<=1e6,问经过操作后偏差值最小为多少? 以及对应的操作次数.


操作一次,循环右移一次,最多n种不同排列.令d[i]为i次操作后的偏差值.


现在考虑p[k]在d[i]中的贡献.
d[0]+=|p[k]-k| d[1]+=|p[k]-(k+1)| ... d[n-k]+=|p[k]-n|,d[n-k+1]+=|p[k]-1|..d[n-1]+=|p[k]-(k-1)|.


当p[k]>=k时 观察p[k]的贡献. (在数轴上画下k,p[k]位置)
对d[0<=i<=p[k]-k] 都是加上p[k]-(k+i).
对d[p[k]-k<i<=n-k] 都是加上(k+i)-p[k].
对d[n-k+i]加上p[k]-i [0<i<k].


上面三个操作 都是区间[l,r]同时加上某个等差数列 (类似加上相同的数,然后差分).
区间[l,r] 加上k*(x-L)+b. 设置两个数列d[i],f[i].


对d[l]+=b,d[r+1]-=b.剩下k*(x-L).
对f[l+1]+=k. f[r+1]-=k 对f求一次前缀和,

然后还要在d[r+1]减去(r-l+1)个k表示该等差数列的结束.

然后对d[i]加上f[i]的前缀和,d[i]在加上本身的前缀和即可得到最后真正的d[i].(....)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e6+20;
ll n,p[N],d[N],f[N],pre[N];
void add(ll k,ll b,int l,int r)
{
	if(l>r)
		return;
	d[l]+=b,d[r+1]-=b;
	f[l+1]+=k,f[r+1]-=k;
	d[r+1]-=k*(r-l);
}
void solve()
{
	ll sum=0,mn=1e18,pos=0;
	for(int i=0;i<n;i++)
		sum+=f[i],d[i]+=sum;
	for(int i=0;i<n;i++)
	{
		d[i]+=d[i-1];
		if(d[i]<mn)
			mn=d[i],pos=i;
	}
	cout<<mn<<' '<<pos<<endl;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%I64d",&p[i]);
	for(int i=1;i<=n;i++)
	{
		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);
		}
	}
	solve();
	return 0;
}