BJ模拟 排列【状压dp+矩阵快速幂】

题目大意：

求长度为$n$，且满足对于任意$i，|p_i-i|\le k$的排列数。$n\le 1e9，k\le 5$

解题思路：

由于$k$很小，很容易想到记录一个$2k+1$位的状态表示当前位还有那些位置可以用，再用矩阵快速幂优化，但还是会TLE。

注意到我们每一步转移恰好增加一个占用，至多减少一个占用，所以合法状态内1的个数是单调不减的，又由于初末状态1的个数都是$k+1$（不存在的位置都是1），所以合法状态内1的个数都是$k+1$，就最多只有$\binom{11}{6}=462$种了，重标号后再dp即可过。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c<'0'||c>'9')&&(c!='-');c=getchar());
    if(c=='-')f=-1,c=getchar();
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=500,mod=1e9+7;
int n,k,m,sta[N],id[1<<11],bin[1<<11];
struct matrix
{
    ll a[N][N];
    matrix(){memset(a,0,sizeof(a));}
    inline friend matrix operator * (const matrix &A,const matrix &B)
    {
        matrix res;
        for(int i=1;i<=m;i++)
            for(int k=1;k<=m;k++)if(A.a[i][k])
                for(int j=1;j<=m;j++)
                    res.a[i][j]=(res.a[i][j]+A.a[i][k]*B.a[k][j])%mod;
        return res;
    }
    inline friend matrix Pow(matrix A,int b)
    {
        matrix res;
        for(int i=1;i<=m;i++)res.a[i][i]=1;
        for(;b;b>>=1,A=A*A)if(b&1)res=res*A;
        return res;
    }
};
void solve()
{
    m=0;int S=0,mx=1<<(2*k+1);
    for(int i=0;i<=k;i++)S|=(1<<i);
    for(int i=0;i<mx;i++)id[i]=0;
    for(int i=0;i<mx;i++)if(bin[i]==k+1)sta[id[i]=++m]=i;
    matrix A,B;A.a[1][id[S]]=1;
    for(int s=0;s<mx;s++)if(id[s])
        for(int i=0;i<=2*k;i++)if(!((s>>1)&(1<<i)))
        {
            int ns=((s>>1)|(1<<i));
            if(id[ns])B.a[id[s]][id[ns]]=1;
        }
    A=A*Pow(B,n);
    cout<<A.a[1][id[S]]<<'\n';
}
int main()
{
    //freopen("lx.in","r",stdin);
    for(int i=1;i<(1<<11);i++)bin[i]=bin[i>>1]+(i&1);
    while(scanf("%d%d",&n,&k)!=EOF)solve();
    return 0;
}