【矩阵快速幂】Codeforces - 1151 - F. Sonya and Informatics

题目链接<http://codeforces.com/contest/1151/problem/F>

---

题意：

给出一串长度位n的01数列，每次随机选两个数字交换，问k次后形成非递减数列的期望是多少。

2≤n≤100,1≤k≤1e9

---

题解：

很明显，就是要把所有的1都放在后面，设dp[i][j]表示第i次操作，有j个1在后面，则有以下几个转移：

1. dp[i][j]由dp[i-1][j]转移：后面和后面交换，前面和前面交换，前面的1和后面的1交换，前面的0和后面的0交换
2. dp[i][j]由dp[i-1][j-1]转移：前面的0和后面的1交换
3. dp[i][j]由dp[i-1][j+1]转移：前面的1和后面的0交换

这样的复杂度是n*k的，把dp数组的第一维消掉，转化成快速幂，就可以log(k)*n^3做出来了

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=1e2+7;
const ll mod=1e9+7;
ll n,k;
ll qpow(ll a,ll b){
    ll res=1;
    while(b){
        if(b&1) res=res*a%mod;
        b>>=1;a=a*a%mod;
    }
    return res;
}
void mul(ll a[N][N],ll b[N][N]){
    ll c[N][N];
    for(int i=0;i<n;i++){
        for(int j=0;j<n;j++){
            c[i][j]=0;
            for(int k=0;k<n;k++){
                if(a[i][k]>0&&b[k][j]>0)
                c[i][j]=(c[i][j]+a[i][k]*b[k][j]%mod)%mod;
            }
        }
    }
    memcpy(a,c,sizeof(c));
}
ll a[N][N];
ll b[N][N];
ll qpow(ll x){
    while(x){
        if(x&1) mul(a,b);
        x>>=1;mul(b,b);
    }
}
int ar[N];

int main()
{
    int t0=0,t1=0,tmp=0;
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++){
        scanf("%d",&ar[i]);
        if(ar[i]) t1++;
        else t0++;
    }
    if(t0==0||t1==0){
        printf("1\n");
        return 0;
    } 
    for(int i=n,j=1;j<=t1;j++,i--){
        if(ar[i]==1) tmp++;
    }
    a[0][tmp]=1;
    
    ll x=n*(n-1)/2;

    b[0][0]=x-t1*t1;b[1][0]=t0-(t1-1);
    for(int i=1;i<t1;i++){
        int j=t1-i;
        b[i][i]=t0*(t0-1)/2+t1*(t1-1)/2+i*(t1-i)+j*(t0-j);
        b[i-1][i]=(t1-i+1)*(t1-i+1);
        b[i+1][i]=(i+1)*(t0-(t1-(i+1)));
    }
    b[t1][t1]=t0*(t0-1)/2+t1*(t1-1)/2;b[t1-1][t1]=1;
    qpow(k);
    ll ans=a[0][t1];
    ans=ans*qpow(qpow(x,k),mod-2)%mod;
    printf("%lld\n",ans);
}