AtCoder 134F

AtCoder 134F

题意：求$1,2\dots ,n$的全排列中有多少满足 $$\sum _{i=1}^n|i-p_i|=K$$
分析：抽象一下，其实是位置与数的匹配，求匹配方案数，假设位置为$A$,数位$B$,建议一一映射求满足条件的映射的个数

1. 搜索复杂度太高，我们分析一下符合$n!$特别大，而$k$只有 $n^2$个状态，所以计数个数非常多
2. $dp$计数状态怎么表示？这里提供一种方法
   $$dp[i][j][j][k]$$
   $i$代表已经匹配到第$i$个
   $j$代表到$i$为止还有j个A和j个B没有匹配
   $k$代表当前已经确定的value是$k$，什么是当前确定的value呢？假设有一个A没有与前i个匹配，那么他肯定只能与后$i+1,i+2,..n$匹配，这样对最后差值的贡献至少加$1$，如果没有与$i+1$匹配，那么下一次同样要加$1$

匹配一对，未匹配数不变$j\to j$，：这是代表A的第i个与B的第j个匹配$(1)$，或者A的第i个与B的未匹配匹配$(j)$,或者B与A的未匹配的匹配$(j)$ 加起来是$2\ast j+1$
$$dp[i][j][k]+=(2\ast j+1)\ast dp[i-1][j][k-2\ast j]$$
匹配两对, 未匹配数减1 $j+1\to j$，：A与j+1个未匹配的匹配$(j+1)$,B与j+1个未匹配$j+1$,乘起来$(j+1)\ast (j+1)$
$$dp[i][j][k]+=(j+1)\ast (j+1)\ast p[i-1][j+1][k-2\ast j]$$
不匹配，未匹配数加1，$j-1\to j$: $1$
$$dp[i][j][k]+=dp[i-1][j-1][k]$$

//
#include <bits/stdc++.h>
using namespace std;
const int mod = 1e9 + 7;
int dp[50+1][50+1][2500+10];// dp[i][j][k] 已经匹配到前i个，有j个没有匹配，当前已知的value是k
void Add(int &a,int b){
	a += b;
	if(a >= mod)
		a -= mod;
}
int main(void)
{
    int N,K;
    cin>>N>>K;
    dp[0][0][0] = 1;
    for(int i  = 1;i <= N;++i){
    	for(int j = 0;j <= i; ++j){
    		for(int k = 2*j;k <= K; ++k){
    			int &a = dp[i][j][k];
    			Add(a,1ll*(2*j+1)*dp[i-1][j][k-2*j]%mod);// i1,i2 匹配，或者i1与一个未匹配的匹配，或者i2与一个没匹配的匹配,没有匹配的个数不变
    			Add(a,1ll*(j+1)*(j+1)*dp[i-1][j+1][k-2*j]%mod);// i1,i2 匹配一个非 i2,i1 的，未匹配的数量-1
    			if(j)
    				Add(a,dp[i-1][j-1][k-2*j]);// i1,i2 不匹配，未匹配的数量+1
    		}
    	}
    }
    cout<<dp[N][0][K]<<endl;
   return 0;
}