经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » C » 查看文章
洛谷P3953 逛公园(dp 拓扑排序)
来源:cnblogs  作者:自为风月马前卒  时间:2018/11/1 9:38:58  对本文有异议

题意

题目链接

Sol

去年考NOIP的时候我好像连最短路计数都不会啊qwq。。

首先不难想到一个思路,\(f[i][j]\)表示到第\(i\)个节点,与最短路之差长度为\(j\)的路径的方案数

首先把每个节点的最短路求出来

转移的时候按拓扑序(也就是按距离从小到大排序)转移一下

然而有\(0\)边的时候会挂掉,原因是会有dis相同的时候,这时候单按dis排序会无法判断转移方向

一种方案是直接把所有\(0\)边加入到新图中,拓扑排序一遍。得到第二关键字

同时判断一下\(0\)

  1. #include<bits/stdc++.h>
  2. #define Pair pair<int, int>
  3. #define MP make_pair
  4. #define fi first
  5. #define se second
  6. using namespace std;
  7. const int MAXN = 100001, INF = 1e9 + 7;
  8. inline int read() {
  9. char c = getchar(); int x = 0, f = 1;
  10. while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
  11. while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
  12. return x * f;
  13. }
  14. int T, N, M, K, mod, f[MAXN][51], dis1[MAXN], disn[MAXN], vis[MAXN], inder[MAXN], id[MAXN];
  15. vector<Pair> v[MAXN], rv[MAXN];
  16. vector<pair<pair<int, int>, int> > P;
  17. vector<int> E[MAXN];
  18. void init() {
  19. memset(f, 0, sizeof(f));
  20. memset(inder, 0, sizeof(inder));
  21. P.clear();
  22. f[1][0] = 1;
  23. N = read(); M = read(); K = read(); mod = read();
  24. for(int i = 1; i <= N; i++) v[i].clear(), E[i].clear();
  25. for(int i = 1; i <= M; i++) {
  26. int x = read(), y = read(), z = read();
  27. v[x].push_back(MP(y, z)); rv[y].push_back(MP(x, z));
  28. if(!z) E[x].push_back(y), inder[y]++;
  29. }
  30. }
  31. void add(int &x, int y) {
  32. if(x + y < 0) x = (x + y + mod);
  33. else x = (x + y >= mod ? x + y - mod : x + y);
  34. }
  35. int Topsort() {
  36. int cnt = 0; queue<int> q;
  37. for(int i = 1; i <= N; i++) if(!inder[i]) id[i] = ++cnt, q.push(i);
  38. while(!q.empty()) {
  39. int p = q.front(); q.pop();
  40. for(int i = 0, to; i < E[p].size(); i++)
  41. if(!(--inder[to = E[p][i]])) q.push(to), id[to] = ++cnt;
  42. }
  43. for(int i = 1; i <= N; i++) if(inder[i] && (dis1[i] + disn[i] <= dis1[N] + K)) return -1;
  44. return 0;
  45. }
  46. void Dij(int bg, int *dis) {
  47. priority_queue<Pair> q; q.push(MP(0, bg));
  48. memset(vis, 0, sizeof(vis));
  49. for(int i = 1; i <= N; i++) dis[i] = INF + 1; dis[bg] = 0;
  50. while(!q.empty()) {
  51. if(vis[q.top().se]) {q.pop(); continue;}
  52. int p = q.top().se; q.pop(); vis[p] = 1;
  53. if(bg == 1) {
  54. for(int i = 0, to; i < v[p].size(); i++)
  55. if(dis[to = v[p][i].fi] > dis[p] + v[p][i].se)
  56. dis[to] = dis[p] + v[p][i].se, q.push(MP(-dis[to], to));
  57. } else {
  58. for(int i = 0, to; i < rv[p].size(); i++)
  59. if(dis[to = rv[p][i].fi] > dis[p] + rv[p][i].se)
  60. dis[to] = dis[p] + rv[p][i].se, q.push(MP(-dis[to], to));
  61. }
  62. }
  63. }
  64. int solve() {
  65. Dij(1, dis1);
  66. Dij(N, disn);
  67. if(Topsort() == -1) return -1;
  68. for(int i = 1; i <= N; i++) P.push_back(MP(MP(dis1[i], id[i]), i));
  69. sort(P.begin(), P.end());
  70. f[1][0] = 1;
  71. for(int j = 0; j <= K; j++) {
  72. for(int i = 0; i < N; i++) {
  73. int x = P[i].se;
  74. for(int k = 0; k < v[x].size(); k++) {
  75. int to = v[x][k].fi, w = v[x][k].se, ps = dis1[x] + j + w - dis1[to];
  76. if(ps <= K) add(f[to][ps], f[x][j]);
  77. }
  78. }
  79. }
  80. int ans = 0;
  81. for(int i = 0; i <= K; i++) add(ans, f[N][i]);
  82. return ans;
  83. }
  84. int main() {
  85. T = read();
  86. while(T--) {
  87. init();
  88. printf("%d\n", solve());
  89. }
  90. return 0;
  91. }
  92. /*
  93. */
 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号