rockethon2015 G2題 Inversions problem 概率dp

題意:給定n,k。k次操作,每次等概率將一個區間翻轉,問最後逆序數對的期望。

思路:設dp[i][j]表示a[i]在a[j]前面的概率。每次枚舉翻轉的區間,更新dp[i][j],複雜度爲O(n^4×k)。詳見代碼:

/*********************************************************
  file name: G.cpp
  author : kereo
  create time:  2015年02月09日 星期一 11時32分13秒
*********************************************************/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int sigma_size=26;
const int N=100+50;
const int MAXN=100000+50;
const int inf=0x3fffffff;
const double eps=1e-8;
const int mod=1000000000+7;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
#define PII pair<int, int>
#define mk(x,y) make_pair((x),(y))
int n,k;
int a[N];
double dp[N][N],res[N][N];
int main(){
    while(~scanf("%d%d",&n,&k)){
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                dp[i][j]=1;
        while(k--){
            memcpy(res,dp,sizeof(dp));
            memset(dp,0,sizeof(dp));
            for(int l=1;l<=n;l++)
                for(int r=l;r<=n;r++){
                    for(int i=1;i<=n;i++)
                        for(int j=i+1;j<=n;j++){
                            int x=i,y=j;
                            if(l<=x && x<=r) x=l+r-x;
                            if(l<=y && y<=r) y=l+r-y;
                            if(x>y)
                                swap(x,y);
                            if(l<=x && y<=r) //在區間裏面,則必然翻轉,所以(1-res[i][j])
                                dp[x][y]+=(1-res[i][j])*2.0/(n+1)/n;
                            else 
                                dp[x][y]+=res[i][j]*2.0/(n+1)/n;  
                        }
                }
        }
        double ans=0;
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                if(a[i]>a[j])
                    ans+=dp[i][j];
                else 
                    ans+=1.0-dp[i][j];
        printf("%.10f\n",ans);
    }
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章