牛客挑战赛32 D放物品（思维+贡献+错排）

D 放物品

题意：

输入$T(10)$组数据，每组数据，输入$n(n<=1000)$,接下来一行输入$n$个数，表示一组$n$的排列$p[i]$，现可将$1-n$任意排列使$i$不出现在$p[i]$位置上，对于任意的排列，每个排列的每一对逆序对$i<j,pos[i]>pos[j]$它的贡献为$(j-i)\ast (pos[i]-pos[j])$,求所有符合要求的排列的贡献和。

题解：

考虑对没两个数进行分析就好，先枚举两个数$x<y$,$x,y$不能在的位置为$ix,iy$,当$x,y$都没取到$ix,iy$,那么就剩下$n-2$个元素(其中有两个数不等(即$ix,iy$没有与之相对应的元素)的错排，于完全错排算法类似，$A_{n-2}^{n-2}-C_{n-4}^1{A}_{n-3}^{n-3}+C_{n-4}^2{A}_{n-4}^{n-4}\dots$,还有要算出$\sum (pos[x]-pos[y])$再乘上$y-x$还有方案数就好。
以上是一种情况，还有$x,y$其中有一个占有$ix,iy$，错排变成有一个元素不一样。
$A_{n-2}^{n-2}-C_{n-3}^1{A}_{n-3}^{n-3}+C_{n-3}^2{A}_{n-4}^{n-4}\dots$
最后一种情况是$ix,iy$都取的情况，它就是标准错排。
$A_{n-2}^{n-2}-C_{n-2}^1{A}_{n-3}^{n-3}+C_{n-2}^2{A}_{n-4}^{n-4}\dots$

代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef long long ll;
const int mod=998244353;
const int N=1009;
int n,a[N],f[N],sum[N],ssum[N],fac[N],inv[N];
int M0,M1,M2;
int get(int x,int y){
    int ix=f[x],iy=f[y];
    ll cnt0=0,cnt1=0,cnt2=0;
    cnt0=(ssum[n-1]-sum[n-ix]-sum[n-iy]+4ll*mod-sum[ix-1]-sum[iy-1]+abs(ix-iy))%mod;
    if(iy<ix){
        cnt1=(sum[iy-1]+sum[n-ix])%mod;
    }
    if(iy>ix){
        cnt2=iy-ix;
        cnt1=(sum[n-ix]+sum[iy-1]-2*cnt2+2*mod)%mod;
    }
    return (cnt0*M0%mod+cnt1*M1%mod+cnt2*M2%mod)%mod*(y-x)%mod;
}
int C(int x,int y){
    return 1ll*fac[x]*inv[y]%mod*inv[x-y]%mod;
}
void getM(){
    int nn=n-2;
    M0=0;
    for(int i=0;i<=nn-2;i++){
        if(i&1)M0=(1ll*M0-1ll*fac[nn-i]*C(nn-2,i))%mod;
        else M0=(1ll*M0+1ll*fac[nn-i]*C(nn-2,i))%mod;
    }
    M0=(M0+mod)%mod;

    M1=0;
    for(int i=0;i<=nn-1;i++){
        if(i&1)M1=(1ll*M1-1ll*fac[nn-i]*C(nn-1,i))%mod;
        else M1=(1ll*M1+1ll*fac[nn-i]*C(nn-1,i))%mod;
    }
    M1=(M1+mod)%mod;

    M2=0;
    for(int i=0;i<=nn;i++){
        if(i&1)M2=(1ll*M2-1ll*fac[nn-i]*C(nn,i))%mod;
        else M2=(1ll*M2+1ll*fac[nn-i]*C(nn,i))%mod;
    }
    M2=(M2+mod)%mod;
}
void solve(){
    getM();
  //  cout<<M0<<" "<<M1<<" "<<M2<<endl;
    int ans=0;
    for(int i=1;i<=n;i++)
      for(int j=i+1;j<=n;j++){
          ans=(ans+get(i,j))%mod;
      }
    cout<<ans<<endl;
}
void init(){
    for(int i=1;i<N;i++)sum[i]=sum[i-1]+i,ssum[i]=(ssum[i-1]+sum[i])%mod;
    fac[0]=fac[1]=inv[0]=inv[1]=1;
    for(int i=2;i<N;i++)fac[i]=1ll*fac[i-1]*i%mod,inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
    for(int i=2;i<N;i++)inv[i]=1ll*inv[i]*inv[i-1]%mod;
}
int t;
signed main(){
    init();
    freopen("tt.in","r",stdin),freopen("tt.out","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
    cin>>t;
    while(t--){
       cin>>n;
       for(int i=1;i<=n;i++)cin>>a[i],f[a[i]]=i;
       if(n==1){cout<<0<<endl;continue;}
       solve();
    }
    return 0;
}