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