[NOIP2017] 逛公园
题目大意:
给定一张图,询问长度 不超过1到n的最短路长度加k 的1到n的路径 有多少条。
数据范围: 点数\(n \le 10^5\) ,边数\(m \le 2*10^5\)
题目解法
两个月后再看也不是太难,自己就能独立思考出来。
首先是判-1的问题,显然能产生-1的只有0环。
所以把0环都找出来,
然后检查一下\(dis[\)\(1\),环\(]\) + \(dis[\)环,\(n]\) 是否小于等于 \(dis[1,n]+K\)即可。
如果不是无限路径的话,也比较套路了。直接把距离扔到\(DP\)维数中肯定不现实。
所以设\(f[ u ][ d ]\)表示从1到u,长度为\(dis[1,u]\)+\(d\) 的路径有多少条。
我们假设\(u\) --> \(v\),\(d_u = rest'\) , \(d_v = rest\) , 那么有如下关系:
\[rest' + dis[1 , u ] + t[ i ].lg\ \ =\ \ rest + dis[1 , v]\]
转移:\(f[v][rest] = \sum f[ u ][rest']\) , 初值\(f[1][0] = 1\)
细节比较多 , 求 \(0\)环 判 -1 可以用\(Tarjan\)做,只走边权为0的边即可。
然后\(DP\)的时候转移顺序不好处理 , 所以记忆化搜索即可。
实现代码:
注:记忆化搜索倒着搜比较方便就倒着搜了。
#include<bits/stdc++.h>
#define RG register
#define IL inline
#define os 55
#define _ 200005
#define INF 1000000007
using namespace std;
IL int gi(){
RG int data = 0 , m = 1; RG char ch = 0;
while(ch != '-' && (ch < '0' || ch > '9') ) ch = getchar();
if( ch == '-' ) { ch = getchar(); m = 0; }
while(ch >= '0' && ch <= '9'){data = (data << 1) + (data << 3) + (ch ^ 48); ch = getchar();}
return ( m ) ? data : -data;
}
int Case,N,M,K,P,zero_res,oo,top,cnt,ans,init[_];
int dis[_][2],dp[_][os],dfn[_],low[_],stk[_],tmp[_],hd[_];
struct Road{int to , next , w ; }t[2*_][ 2 ] ; int head[ _ ][ 2 ];
bool vis[_]; queue<int>Q;
IL void spfa(RG int S , RG int id){
for(RG int i = 1; i <= N; i ++)dis[ i ][id] = INF ;
vis[ S ] = true; Q.push( S ) ; dis[ S ][id] = 0;
while(!Q.empty()){
RG int u = Q.front(); Q.pop();
for(RG int i = head[ u ][id] ; i ; i = t[ i ][id].next){
RG int v = t[ i ][id].to ;
if(dis[ v ][id] > dis[ u ][id] + t[ i ][id].w){
dis[ v ][id] = dis[ u ][id] + t[ i ][id].w ;
if(! vis[ v ] ) Q.push( v ) , vis[ v ] = true;
}
}vis[ u ] = false;
}return;
}
IL void Tarjan( RG int u ){
stk[ ++ top ] = u;
dfn[ u ] = low[ u ] = ++ oo ; init[ u ] = true;
for(RG int i = head[ u ][0] ; i ; i = t[ i ][0].next){
if(t[ i ][0].w != 0)continue; RG int v = t[ i ][0].to;
if( !dfn[ v ] )
Tarjan( v ) , low[ u ] = min(low[ u ] , low[ v ]) ;
else if(init[ v ])low[ u ] = min(low[ u ] , dfn[ v ]) ;
}
if(low[ u ] == dfn[ u ]){
RG int e , ct = 0;
while(1){
e = stk[ top ] ; top --;
init[e] = false; tmp[ ++ct ] = e;
if(e == u || !top) break;
}
if(ct >= 2)
zero_res = min( dis[ u ][ 0 ] + dis[ u ][ 1 ] , zero_res) ;
}return;
}
IL int DP( RG int u , RG int rest ) {
if( dp[ u ][ rest ] )return dp[ u ][ rest ] ;
for(RG int i = head[ u ][ 1 ] ; i ; i = t[ i ][ 1 ].next){
RG int v = t[ i ][ 1 ].to , d;
d = rest + dis[ u ][ 0 ] - dis[ v ][ 0 ] - t[ i ][ 1 ].w ;
if( d < 0 )continue;
dp[ u ][ rest ] = ( dp[ u ][ rest ] + DP( v , d ) ) % P;
}return dp[ u ][ rest ] ;
}
//v-->u : rest' + dis[v][0] + t[i].w = rest + dis[u][0]
int main(){
freopen("2017park.in" , "r" , stdin);
freopen("2017park.out" , "w" , stdout);
Case = gi();
while(Case -- ){
N = gi(); M = gi(); K = gi(); P = gi();
for(RG int i = 1; i <= N; i ++)head[ i ][ 0 ] = 0;
for(RG int i = 1; i <= N; i ++)head[ i ][ 1 ] = 0;
for(RG int i = 1; i <= N; i ++)dfn[ i ] = low[ i ] = 0;
cnt = 0;
for(RG int i = 1 , u , v , c; i <= M; i ++){
u = gi(); v = gi(); c = gi(); ++ cnt ;
t[ cnt ][ 0 ] = ( Road ) { v , head[ u ][ 0 ] , c } ;
head[ u ][ 0 ] = cnt ;
t[ cnt ][ 1 ] = ( Road ) { u , head[ v ][ 1 ] , c } ;
head[ v ][ 1 ] = cnt ;
//0 : u --> v (oder) // 1 : v --> u (dder)
}
spfa(1 , 0) ; spfa(N , 1) ;
zero_res = INF ;
for(RG int i = 1; i <= N ; i ++) if( ! dfn[ i ] ) Tarjan( i ) ;
if(zero_res <= dis[ N ][ 0 ] + K){ puts("-1") ; continue; }
for(RG int i = 1; i <= N; i ++)
for(RG int j = 0; j <= K; j ++)
dp[ i ][ j ] = 0;
dp[ 1 ][ 0 ] = 1; ans = 0;
for(RG int delta = 0; delta <= K; delta ++)
ans = ( ans + DP( N , delta ) ) % P;
cout << ans << endl;
}return 0;
}
原文地址:https://www.cnblogs.com/GuessYCB/p/8463483.html
时间: 2024-11-05 22:54:48