POJ-5001-Walk

這個題我本來想用記憶化,因爲最開始想過這種DP的,發現複雜度比較高,但後來我用記憶化交上去無奈TLE,然後賽後他們居然說是用這樣的DP寫的,無奈,重寫~

        重新認識杭電的性能了,其實比較簡單吧,dp[i][j][k]表示第i次到第j個點未經過k的概率,然後看代碼吧~、

代碼:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=51;
const int maxm=maxn*maxn;
int e,head[maxn],pnt[maxm],nxt[maxm];
int n,m,d,cnt[maxn];
double dp[2][maxn][maxn],ans[maxn];
void AddEdge(int u,int v)
{
    pnt[e]=v;nxt[e]=head[u];head[u]=e++;
    pnt[e]=u;nxt[e]=head[v];head[v]=e++;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        e=0;
        memset(head,-1,sizeof(head));
        memset(dp,0,sizeof(dp));
        memset(cnt,0,sizeof(cnt));
        scanf("%d%d%d",&n,&m,&d);
        for(int i=0;i<m;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            AddEdge(u,v);
            cnt[u]++;
            cnt[v]++;
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(i!=j)
                    dp[0][i][j]=1.0/n;
        int pos=1;
        for(int i=1;i<=d;i++,pos^=1)
            for(int j=1;j<=n;j++)
                for(int k=1;k<=n;k++)
                    if(j!=k)
                    {
                        dp[pos][j][k]=0;
                        for(int s=head[j];s!=-1;s=nxt[s])
                            dp[pos][j][k]+=dp[pos^1][pnt[s]][k]/cnt[pnt[s]];
                    }
        memset(ans,0,sizeof(ans));
        pos^=1;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                ans[j]+=dp[pos][i][j];
        for(int i=1;i<=n;i++)
            printf("%.10f\n",ans[i]);

    }
    return 0;
}


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