UOJ-311 逛公园

//好久不见。上上周忙于加试,上周又不幸因病住院,我也很无奈啊。

首先我们注意到k的值非常小,最大也只能达到50,复杂度一定与它有关。

然后又是需要取模的计数问题。考虑dp。


首先是本人写的很丑的非常慢的解法:

首先我们跑dijkstra处理出两个数组dis0,dis1,分别维护从起点和终点到达该点的距离。

dp状态:f[i][j]表示到第i个顶点路径长度比从起点到该点最短路长j的方案数。

转移很简单,对于每一个f[i][j]枚举以i为起点的边,令该边的另一端点为v,权值为d。那么状态转移方程为:

f[v][j+dis0[i]+d-dis0[v]]+=f[i][j]

其中j+dis0[i]就是f[i][j]对应的距离。显然,dis0[i]+d-dis0[v]>=0,因此把j放在外层转移就可以了

吗?

nonono 显然不行,dp值可能是由第二个维度相同的另一个dp值转换过来的。对于这种情况,我们需要对第一个维度进行拓扑排序。

我们需要对全图进行拓扑排序吗?显然不需要。因为有些点根本不会出现在结果方案的路径上(如果一个点到起点的最短路径和到终点的最短路径之和与起点到终点最短路径的差都大于k,即dis0[i]+dis1[i]-dis0[n]>k,那么这个点一定不会被用到)。再者图是可以存在环的,且并不是所有的环都会导致-1的情况。我们可以反复走环。考虑到只有终点与起点距离差等于边权的情况是不能从更小的j转移过来的,我们只需要对这些边的端点进行拓扑排序即可,其他可用的点随意放。而如果拓扑排序无法完成即代表存在0环。这也就是题目中所说的-1的情况。

第一遍做的时候只考虑到了根据到起点最短距离进行排序,但这种做法是无法处理环上的0边的。因为环起点处的一些dp值是可以从终点转移过来的。但终点到起点的距离一定是大于起点的,无法正确转移。

空间是O(nk)的,时间因为是通过边转移,可以达到O(mk),大概是1e7,如果常数写大点还是非常危险的。但仔细想想,首先对于f[i][j]=0的情况显然是不用转移的,可以continue掉。其次我们只需要跑可用的点,因此是跑不到1e7的。再加上题目的时限给到了3s,还是可以基本放心的。

上丑陋无比的代码。因为一开始思路并不是很明确写的有点乱。

#include
using namespace std;

#define pb push_back
#define fi first
#define se second
#define ll long long
#define pq priority_queue
#define mp make_pair
#define pii pair
#define mod 998244353
#define debug(x) cerr<<#x<<"="< edge[2][maxn];
int dis[2][maxn];
int f[maxn][55];

void dijkstra(int x) {
    pq ,greater > q;
    int sta;
    if (x==0) sta=1;
    else sta=n;
    for (int i=1;i<=n;i++) dis[x][i]=2e9+1119;
    dis[x][sta]=0;
    q.push(mp(0,sta));
    while (!q.empty()) {
        int u=q.top().se,d=q.top().fi;
        q.pop();
        if (dis[x][u]!=d) continue;
        for (int i=0;idd+d) {
                dis[x][v]=dd+d;
                q.push(mp(dis[x][v],v));
            }
        }
    }
    return;
}

int deg[maxn];
int tmp[maxn];
bool chosen[maxn];
int bg,ed;

bool check() {
    memset(deg,0,sizeof(deg));
    memset(tmp,0,sizeof(tmp));
    memset(chosen,0,sizeof(chosen));
    for (int i=1;i<=n;i++) 
        for (int j=0;jk) continue;
                (f[v][tmp]+=f[cur][i])%=p;
            }
        }
    int ans=0;
    for (int i=0;i<=k;i++) {
        (ans+=f[n][i])%=p;
    }  
    printf("%d\n",ans);
    return;
}

int main(){
    scanf("%d",&T);
    while (T--) solve();
    return 0;
}

然后介绍另一种做法。大差不差,只不过是那个做法直接把各边的权值改成了经过该边对结果(路径长与最短路的差值)的贡献,即对于原来从ij权值为d的边把其权值修改为dis0[i]+d-dis0[j]。这样无论是dp还是拓扑排序写起来都会清爽很多,不容易出错。

你可能感兴趣的:(算法——动态规划,图论——拓扑排序)