題意：給一張帯權有向圖，求 $1$ 到 $n$ 長度不超過最短路長度 $+k$ 的路徑條數 模 $P$。有無數條輸出 $?1$ 。

$n\le 10^5,m\le 2\times 10^5,k\le 50$，邊權非負

先往最短路圖想

發現 $k$ 很小，考慮硬上 dp。

設 $f(u,k)$ 表示從 $u$ 到 $n$ 多走 $k$ 的路徑數。

轉移枚舉下一步往哪里走。

$$f(u,k)=\sum _{(u,v)\in E}f(v,k?(w(u,v)+di{s}_u?di{s}_v))$$

其中 $di{s}_u$ 為 $1$ 到 $u$ 最短路長度。

在沒有零邊的基礎上，右邊這塊 $w(u,v)+di{s}_u?di{s}_v$ 取 $0$ 當且僅當 $(u,v)$ 在最短路圖上。因為這是個 DAG，所以在最短路圖上跑拓撲排序，然后枚舉 $k$ 更新就可以了。這樣可以拿到 $70$ 分的好成績。

有零邊時可能會出現 $di{s}_n+k$ 內可達的零環，這樣答案為無窮大。如果拓撲排序后點 $i$ 入度不為 $0$ 說明在零環上，此時如果 $di{s}_i+di{s}_i^{\prime }\le di{s}_n+k$ 就輸出 $?1$。（$di{s}^{\prime }$ 為到 $n$ 的最短距離）

否則的話反正都走不到這里，直接把偽的拓撲序繼續往后做。

復雜度 $O(n\log n+nk)$

#include <iostream> #include <cstdio> #include <cstring> #include <cctype> #include <algorithm> #include <utility> #include <queue> #define MAXN 100005 #define MAXM 200005 using namespace std; int n,m,k,P; inline int add(const int& x,const int& y){return x+y>=P? x+y-P:x+y;} inline int read() {int ans=0;char c=getchar();while (!isdigit(c)) c=getchar();while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();return ans; } struct edge{int u,v,w;}e[MAXM]; int head[MAXN],nxt[MAXM],cnt; inline void addnode(int u,int v,int w) {e[++cnt]=(edge){u,v,w};nxt[cnt]=head[u];head[u]=cnt; } template<typename T,typename cmp=less<T> > struct heap {T val[MAXM];T* end;heap(){end=val;}inline bool empty(){return end==val;}inline void push(const T& v){*(end++)=v;push_heap(val,end,cmp());}inline T pop(){pop_heap(val,end,cmp());return *(--end);} }; typedef pair<int,int> pi; heap<pi,greater<pi> > q; int dis[MAXN],disn[MAXN],deg[MAXN]; inline void dij(int* dis,int s=1) {memset(dis,0x3f,sizeof(disn));dis[s]=0;q.push(make_pair(0,s));while (!q.empty()){int u=q.pop().second;for (int i=head[u];i;i=nxt[i])if (dis[u]+e[i].w<dis[e[i].v])dis[e[i].v]=dis[u]+e[i].w,q.push(make_pair(dis[e[i].v],e[i].v));} } int lis[MAXN],vis[MAXN],tim; void dfs(int u) {vis[u]=1;for (int i=head[u];i;i=nxt[i])if (dis[u]+e[i].w==dis[e[i].v]){++deg[e[i].v];if (!vis[e[i].v]) dfs(e[i].v);} } inline void topsort() {queue<int> q;tim=0;q.push(1);while (!q.empty()){int u=q.front();q.pop();lis[++tim]=u;for (int i=head[u];i;i=nxt[i])if (dis[u]+e[i].w==dis[e[i].v]&&(--deg[e[i].v]==0))q.push(e[i].v);} } int f[55][MAXN],u[MAXM],v[MAXM],w[MAXM]; int main() {for (int T=read();T;T--){cnt=tim=0;memset(head,0,sizeof(head));memset(nxt,0,sizeof(nxt));memset(deg,0,sizeof(deg));memset(vis,0,sizeof(vis));n=read(),m=read(),k=read(),P=read();for (int i=1;i<=m;i++) u[i]=read(),v[i]=read(),w[i]=read(),addnode(v[i],u[i],w[i]);dij(disn,n);cnt=0;memset(head,0,sizeof(head));memset(nxt,0,sizeof(nxt)); for (int i=1;i<=m;i++) addnode(u[i],v[i],w[i]);dij(dis);dfs(1);topsort();int ans=0;for (int i=1;i<=n;i++) if (deg[i]&&dis[i]+disn[i]<=dis[n]+k) {puts("-1");goto end;}memset(f,0,sizeof(f));f[0][n]=1;for (int t=0;t<=k;t++)for (int x=tim;x>=1;x--){int u=lis[x];for (int i=head[u];i;i=nxt[i]){int del=e[i].w-dis[e[i].v]+dis[u];if (t>=del) f[t][u]=add(f[t][u],f[t-del][e[i].v]);}}for (int i=0;i<=k;i++) ans=add(ans,f[i][1]);printf("%d\n",ans);end:;}return 0; }

總結

以上是生活随笔為你收集整理的【NOIP2017】逛公园【最短路DAG】【dp】【拓扑排序】的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。