Codeforces1151 F. Sonya and Informatics（概率dp+矩阵快速幂优化）

题意：

给定长度为n的序列，其中每个元素为0或者1，
给定一个k，接下来会进行k次操作，每次操作：
随机选择两个不同的下标i和j，交换a(i)和a(j)，
问使得数组非降序(即00001111)的概率是多少。
答案对1e9+7取模

数据范围：n<=100，k<=1e9

解法：

设x为序列中0的个数,
最后的目标是使得前x个数全部都是0,
d[i][j]表示i次操作,序列前x个数中有j个0的概率,
d[i][j]可以由d[i-1][j-1],d[i-1][j],d[i-1][j+1]转移得来,
考虑转移的概率系数:
d[i-1][j-1]转移到d[i][j]方案数为(x-(j-1))*(x-(j-1)),
意思是从[1,x]中选出一个1,从[x+1,n]中选出一个0,
设当前方案数为p,因为总方案数为n*(n-1)/2,设其为tot,
那么概率为p/tot,其他转移的概率系数同理.

因此可以发现转移系数只和j有关,对于固定的j,系数是固定的,
那么可以构造出系数矩阵,用矩阵快速幂优化dp的转移.

答案为d[k][x]

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=105;
const int mod=1e9+7;
int a[maxm];
struct Node{
    int a[maxm][maxm];
    Node operator*(const Node& x)const{
        Node ans;
        for(int i=0;i<maxm;i++)for(int j=0;j<maxm;j++)ans.a[i][j]=0;
        for(int k=0;k<maxm;k++){
            for(int i=0;i<maxm;i++){
                for(int j=0;j<maxm;j++){
                    ans.a[i][j]=(ans.a[i][j]+a[i][k]*x.a[k][j])%mod;
                }
            }
        }
        return ans;
    }
};
int ppow(int a,int b,int mod){
    int ans=1%mod;a%=mod;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
Node Node_pow(Node a,int b){
    Node ans;
    for(int i=0;i<maxm;i++)for(int j=0;j<maxm;j++)ans.a[i][j]=(i==j);
    while(b){
        if(b&1)ans=ans*a;
        a=a*a;
        b>>=1;
    }
    return ans;
}
signed main(){
    int n,k;
    cin>>n>>k;
    int x=0;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(!a[i])x++;
    }
    //构造系数矩阵
    Node base;
    for(int i=0;i<maxm;i++)for(int j=0;j<maxm;j++)base.a[i][j]=0;
    for(int i=0;i<=x;i++){//计算base的第i列
        base.a[i][i]+=x*(x-1)/2;//0和0换
        base.a[i][i]+=(n-x)*(n-x-1)/2;//1和1换
        base.a[i][i]+=i*(x-i)+(x-i)*(n-x-x+i);//[1,x]或[x+1,n]内部的0和1换.
        base.a[i][i]%=mod;
        if(i>0){
            base.a[i-1][i]+=(x-(i-1))*(x-(i-1));//[1,x]选出一个1,[x+1,n]选出一个0
            base.a[i-1][i]%=mod;
        }
        if(i<x){
            base.a[i+1][i]+=(i+1)*(n-x-x+(i+1));//[1,x]选出一个0,[x+1,n]选出一个1
            base.a[i+1][i]%=mod;
        }
    }
    int inv=ppow(n*(n-1)/2,mod-2,mod);
    for(int i=0;i<=x;i++){
        for(int j=0;j<=x;j++){
            base.a[i][j]*=inv;
            base.a[i][j]%=mod;
        }
    }
    //
    Node temp=Node_pow(base,k);
    int c=0;
    for(int i=1;i<=x;i++)if(!a[i])c++;
    Node ans;
    for(int i=0;i<maxm;i++)for(int j=0;j<maxm;j++)ans.a[i][j]=0;
    ans.a[0][c]=1;
    ans=ans*temp;
    cout<<ans.a[0][x]<<endl;
    return 0;
}

---