[NOIP2017]逛公园

Description
给出一张n个点m条边的有向图，边权为非负整数。求满足路径长度小于等于1到n最短路+k的1到n的路径条数模p，如果有无数条则输出−1。

Input
第一行包含一个整数T,代表数据组数。
接下来T组数据，对于每组数据：第一行包含四个整数N,M,K,P，每两个整数之间用一个空格隔开。
接下来M行，每行三个整数ai,bi,ci，代表编号为ai,bi的点之间有一条权值为ci的有向边，每两个整数之间用一个空格隔开。

Output
输出文件包含T行，每行一个整数表示答案。

Sample Input
2
5 7 2 10
1 2 1
2 4 0
4 5 2
2 3 2
3 4 1
3 5 2
1 5 3
2 2 0 10
1 2 0
2 1 0

Sample Output
3
-1

---

首先求单源最短路，然后设\(f[i][k]\)表示当前到i号点，所经过的路程比最短路多了k。那么转移显然为\(f[x][dis[son]-dis[x]+v-val[p]]\Rightarrow f[son][v]\)，因此这个转移肯定是枚举反向边进行转移。然后我们可以对\(f\)进行记搜，这样便可以省略拓扑判零环。

/*program from Wolfycz*/
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define inf 0x7f7f7f7f
using namespace std;
typedef long long ll;
typedef unsigned int ui;
typedef unsigned long long ull;
inline int read(){
    int x=0,f=1;char ch=getchar();
    for (;ch<'0'||ch>'9';ch=getchar())  if (ch=='-')    f=-1;
    for (;ch>='0'&&ch<='9';ch=getchar())    x=(x<<1)+(x<<3)+ch-'0';
    return x*f;
}
inline void print(int x){
    if (x>=10)  print(x/10);
    putchar(x%10+'0');
}
const int N=1e5,M=2e5;
int n,m,K,MOD,Ans;
struct Link{
    int pre[M+10],now[N+10],child[M+10],val[M+10],tot;
    void init(){tot=0,memset(now,0,sizeof(now));}
    void insert(int x,int y,int z){pre[++tot]=now[x],now[x]=tot,child[tot]=y,val[tot]=z;}
}Frw,Bck;//Forward,Back
int h[N+10],dis[N+10],f[N+10][55];
bool vis[N+10],can[N+10][55],flag;
void SPFA(int x){
    int head=0,tail=1;
    memset(dis,63,sizeof(dis));
    h[1]=x,dis[x]=0,vis[x]=1;
    while (head!=tail){
        if (++head>N)   head=1;
        int Now=h[head];
        for (int p=Frw.now[Now],son=Frw.child[p];p;p=Frw.pre[p],son=Frw.child[p]){
            if (dis[son]>dis[Now]+Frw.val[p]){
                dis[son]=dis[Now]+Frw.val[p];
                if (!vis[son]){
                    if (++tail>N)   tail=1;
                    vis[h[tail]=son]=1;
                }
            }
        }
        vis[Now]=0;
    }
}
int dfs(int x,int v){
    if (~f[x][v])   return f[x][v];
    can[x][v]=1,f[x][v]=(x==1&&!v)?1:0;
    //到了(1,0)不要立刻停止，因为有可能出现全图都是0边的情况，直接退出就会挂
    for (int p=Bck.now[x],son=Bck.child[p];p;p=Bck.pre[p],son=Bck.child[p]){
        int tmp=dis[x]+v-dis[son]-Bck.val[p];
        if (tmp>=0){
            if (can[son][tmp])  flag=1;
            f[x][v]=(f[x][v]+dfs(son,tmp))%MOD;
        }
    }
    can[x][v]=0;
    return f[x][v];
}
void init(){
    flag=0,Ans=0;
    Frw.init(),Bck.init();
    memset(f,255,sizeof(f));
    n=read(),m=read(),K=read(),MOD=read();
}
int main(){
    for (int Data=read();Data;Data--){
        init();
        for (int i=1;i<=m;i++){
            int x=read(),y=read(),z=read();
            Frw.insert(x,y,z);
            Bck.insert(y,x,z);
        }
        SPFA(1);
        for (int i=0;i<=K;i++)  Ans=(Ans+dfs(n,i))%MOD;
        printf("%d\n",flag?-1:Ans);
    }
    return 0;
}

转载于:https://www.cnblogs.com/Wolfycz/p/9744271.html