CF819B Mister B and PR Shifts 思维 + 模拟

link

思路： 涉及绝对值，我们分为两类数来看，第一类是 $p_i-i>0$ 第二类是 $p_i-i<=0$ 。

记zcnt为第一类个数，ztot为第一类的贡献。fcnt为第二类个数，ftot为第二类的贡献。

先不考虑边界情况，对于右移一位 i 会增加1，第一类数 ztot - zcnt，第二类数 ftot + fcnt。当然随着移动会有第一类数变成第二类数的情况，我们用 te 数组记录一下什么时候会改变，让后减去对应的 cnt 即可。

考虑移动的时候当前最后一位数移动到第一位，特判一下就好啦，由于在最后一位一定是第二类数，所以要减去这个数第二类中的贡献。

再就是把数组开大点就好了。

//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid (tr[u].l+tr[u].r>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;

//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }

typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;

const int N=4000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;

int n;
int p[N],te[N];
LL ztot,zcnt,ftot,fcnt;

int main()
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);

    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&p[i]);
        if(p[i]-i>0)
        {
            zcnt++;
            ztot+=p[i]-i;
            te[p[i]-i]++;
        }
        else
        {
            fcnt++;
            ftot+=(i-p[i]);
        }
    }
    LL ans=ftot+ztot,id=0;
    for(int i=1;i<n;i++)
    {
        ztot-=zcnt;
        zcnt-=te[i];
        ftot+=fcnt;
        fcnt+=te[i];
        int x=p[n-i+1];
        ftot-=n+1-x; fcnt--;
        if(x>1) te[x-1+i]++,ztot+=x-1,zcnt++;
        else fcnt++;
        if(ans>ztot+ftot) ans=ztot+ftot,id=i;
    }
    printf("%lld %lld\n",ans,id);



















	return 0;
}
/*

*/