POJ-5001-Walk

1857 단어 dp
이 문제는 제가 기억화를 하려고 했는데 처음에 이런 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
",ans[i]); } return 0; }

좋은 웹페이지 즐겨찾기