UESTC - 1147 求最短路方案数

这道题很是说明了记忆化搜索的重要性
瞎bfs递推半天发现没卵用(也许是姿势不对,但我认为树形或图形dfs明显好写并且很好正确地递推)
参考了别人的写法,总感觉自己的实现能力太弱了
还有题目是1e9+9,送了3WA(一眼1e9+7hhhhh)

/*H E A D*/
int to[maxn<<1],nxt[maxn<<1],cost[maxn<<1],head[maxn],tot;
void init(){
    memset(head,-1,sizeof head);
    tot=0;
} 
void add(int u,int v,int w){
    to[tot]=v;cost[tot]=w;nxt[tot]=head[u];head[u]=tot++;
    swap(u,v);
    to[tot]=v;cost[tot]=w;nxt[tot]=head[u];head[u]=tot++;
}
int dp[maxn];
int dis[maxn];
int n,m;
typedef pair<int,int> P;
void dijkstra(int s){
    memset(dis,oo,sizeof dis);
    priority_queue<P,vector<P>,greater<P> > que;
    que.push(P(s,0));
    dis[s]=0; dp[s]=1;
    while(!que.empty()){
        P p=que.top(); que.pop();
        int u=p.first;
        if(dis[u]<p.second)continue;
        erep(i,u){
            int v=to[i],w=cost[i];
            if(dis[v]>dis[u]+w){
                dis[v]=dis[u]+w;
                que.push(P(v,dis[v]));
            }
        }
    }
}
bool vis[maxn];
bool findzero(int u){
    if(u==1)return 0;
    bool flag=0;
    erep(i,u){
        int v=to[i],w=cost[i];
        if(!vis[i]&&dis[u]==dis[v]+w){
            vis[i]=vis[i^1]=1;
            if(w==0) return 1;
            if(findzero(v)) return 1;
        }
    }
    return 0;
}
int DP(int u){
    if(~dp[u]) return dp[u];
    dp[u]=0;
    erep(i,u){
        int v=to[i],w=cost[i];
        if(!vis[i]&&dis[u]==dis[v]+w){
            vis[i]=1;
            vis[i^1]=1;
            dp[u]=(1ll*dp[u]+DP(v))%mod;
        }
    }
    return dp[u];
}
int main(){
    while(cin>>n>>m){
        init();
        rep(i,1,m){
            int u=read();
            int v=read();
            int w=read();
            add(u,v,w);
        }
        dijkstra(1);
        memset(vis,0,sizeof vis);
        bool flag=findzero(n);
        if(flag) println(-1);
        else{
            memset(dp,-1,sizeof dp);
            memset(vis,0,sizeof vis);
            dp[1]=1;
            println(DP(n));         
        }
    }
    return 0;
}

一组测试用数据

4 5
1 2 2
1 3 1
2 3 1
2 4 1
3 4 2

ans:3

顺便挂一下看着不错的写法:http://blog.csdn.net/code12hour/article/details/52081457

转载于:https://www.cnblogs.com/caturra/p/8448995.html