[hackerrank Random]

题目大意

给你一个n个整数的数组d，有两种操作：
1. 随机一个二元组(l,r)[l < <script type="math/tex" id="MathJax-Element-42"><</script>r]，然后把d[l]和d[r]交换
2. 随机一个二元组(l,r)[l < <script type="math/tex" id="MathJax-Element-43"><</script>r]，然后把区间[l,r]翻转

你要先给d进行a次操作1，再进行b次操作2，然后随机一个二元组(l,r)[l < <script type="math/tex" id="MathJax-Element-44"><</script>r]，得到区间[l,r]元素的和。
给定n,a,b和d[]，求最终得到数字的期望。
2≤n≤1000 a≤$10^9$ b≤10

分析

这道题可以切成三部分：操作1，操作2，还有得到区间和。

最后一步

首先从简单处入手。
假设已经得到了最终的数组d，数组的下标是0—n-1，那么一个数对答案的贡献就是它被区间覆盖的概率。所以容易得出：

$$Ans=\sum_{i=0}^{n-1} d[i]*{\frac{i*(n-i)+n-i-1}{\frac{n(n-1)}{2}}}$$

操作2

接着考虑次简单的部分。
假设已经得到了操作1完成后的数组d，那么设f[k][i]表示进行了k次操作2后的d[i]的期望值。
考虑转移，枚举上一层的j对i的贡献。可以发现，当i≠j时，如果i,j在选择的区间[l,r]里是对称的（即i-l==r-j），那么就会对答案有贡献。计算有多少个区间满足即可。对所有j求完后，剩下没有被计算过的区间个数即是f[k-1][i]转移到f[k][i]的区间个数了。
转移式就不写了，时间复杂度$O(n^2b)$

操作1

最后看操作1。
首先考虑一次操作，设E(x)表示转移一次后第x位的期望。那么有：

$$E(x)=((\sum_{x≠i} d[i]*\frac{1}{\frac{n(n-1)}{2}}+d[x]*(1-\frac{n-1}{\frac{n(n-1)}{2}})$$
$$E(x)=\sum d[i]*\frac{2}{n(n-1)}+d[x]*(1-\frac{2}{n}-\frac{2}{n(n-1)})$$
$\sum d[i]$可以直接求出来，令其为S，它一定是不会随操作而变化的。令E(i,x)表示i次操作后d[x]的期望，发现式子可以变成：
$$E(a,x)=d[x]*(\frac{n-3}{n-1})^a+\sum_{i=0}^{a-1} (\frac{n-3}{n-1})^i*S*\frac{2}{n(n-1)}$$
后面部分是等比数列，得到：
$$E(a,x)=d[x]*(\frac{n-3}{n-1})^a+S*\frac{2}{n(n-1)}*\frac{1-(\frac{n-3}{n-1})^a}{1-\frac{n-3}{n-1}}$$
那么E(a)[]可以用快速幂 $O(nloga)$求出来。

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N=1005;

typedef long long LL;

typedef double db;

int n,a,b,d[N];

db E[N],f[11][N],tmp,s,ans;

db quick(db x,int y)
{
    if (!y) return 1;
    db tmp=quick(x,y>>1);
    tmp*=tmp;
    if (y&1) tmp*=x;
    return tmp;
}

int main()
{
    scanf("%d%d%d",&n,&a,&b);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&d[i]); s+=d[i];
    }
    tmp=(db)(n-3)/(n-1);
    for (int i=1;i<=n;i++)
    {
        E[i]=d[i]*quick(tmp,a)+s*2/(n*(n-1))*((1-quick(tmp,a))/(1-tmp));
    }
    memcpy(f[0],E,sizeof(E));
    tmp=2.0/n/(n-1);
    for (int k=1;k<=b;k++)
    {
        for (int i=1;i<=n;i++)
        {
            f[k][i]+=f[k-1][i];
            for (int j=1;j<i;j++) f[k][i]+=(f[k-1][j]-f[k-1][i])*min(j,n-i+1)*tmp;
            for (int j=i+1;j<=n;j++) f[k][i]+=(f[k-1][j]-f[k-1][i])*min(i,n-j+1)*tmp;
        }
    }
    ans=0;
    for (int i=1;i<=n;i++) ans+=f[b][i]*((i-1)*(n-i+1)+n-i);
    printf("%.6lf\n",ans*tmp);
    return 0;
}