[ZROI160]路径计数 DP+广义容斥

• 50pts
• f[s][t][d][x]表示从s点出发走到x路径长度为d且不经过s和t的方案数

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
using namespace std;
const int N=101;
struct node{int y,n;}e[N*N];
int f[N][N][51][N],v[N][N],lin[N*N],len=0,ans=0,n,m,p,q,x,y,s,t,d;
void read(int x,int y)
{e[++len].y=y,e[len].n=lin[x],lin[x]=len;}
void work(int s,int t){
	f[s][t][0][s]=1;
	rep(d,0,49){
		rep(x,1,n){
			for(int i=lin[x];i;i=e[i].n){
				int y=e[i].y;
				if(y==s||y==t)continue;
				f[s][t][d+1][y]=(f[s][t][d+1][y]+f[s][t][d][x])%p;
			}
		}
	}
}
int main()
{
	scanf("%d%d%d",&n,&m,&p);
	rep(i,1,m){
		scanf("%d%d",&x,&y);
		read(x,y);v[y][x]=1;
	}
	rep(s,1,n)rep(t,1,n)work(s,t);
	scanf("%d",&q);
	while(q--){
		scanf("%d%d%d",&s,&t,&d);
		ans=0;
		rep(i,1,n)if(v[t][i])ans=(ans+f[s][t][d-1][i])%p;
		printf("%d\n",ans);
	}
	return 0;
}

• 100pts
• f[s][t][d]表示从s走到t且中途不经过s，t的长度为d的路径条数 
• g[s][t][d]表示从s到t的长度为d的路径条数
• h[s][t][d]表示从s走到s，且途中不经过s，t长度为d的路径条数
• 答案就是  f[s][t][d]  =  g[s][t][d]-不合法方案数

#include<bits/stdc++.h>
#define rep(i,a,b) 	for(register int i=(a);i<=(b);++i)
#define per(i,a,b) 	for(register int i=(a);i>=(b);--i)
#define mul(a,b) 	1LL*a*b%p
using namespace std;
const int N=102;
int f[N][N][N/2],g[N][N][N/2],h[N][N][N/2],n,m,p,x,y,q,s,t,d;
inline void add(int &a,int b){a+=b;a-= a>=p? p:0;}
inline void sub(int &a,int b){a-=b;a=(a+p)%p;}
inline int read(){
	int num=0;char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))num=num*10+ch-'0',ch=getchar();
	return num;
}
inline void print(int x){ if(x>9)print(x/10);putchar(x%10+'0'); }
int main()
{
	n=read(),m=read(),p=read();
	rep(i,1,m){
		x=read(),y=read();
		f[x][y][1]=g[x][y][1]=1;
	}
	rep(d,2,50){
		rep(k,1,n)rep(s,1,n)rep(t,1,n)add(g[s][t][d],mul(g[s][k][d-1],g[k][t][1]));
		rep(s,1,n){
			rep(t,1,n){
				f[s][t][d]=g[s][t][d];
				rep(k,1,d-1){
					if(s==t)sub(f[s][t][d],mul(f[s][t][k],g[t][t][d-k]));
					else{
						sub(f[s][t][d],mul(f[s][t][k],g[t][t][d-k]));
						sub(f[s][t][d],mul(h[s][t][k],g[s][t][d-k]));
					}
				}
			}
		}
		rep(s,1,n){
			rep(t,1,n){
				h[s][t][d]=g[s][s][d];
				rep(k,1,d-1){
					sub(h[s][t][d],mul(f[s][t][k],g[t][s][d-k]));
					sub(h[s][t][d],mul(h[s][t][k],g[s][s][d-k]));
				}
			}
		}
	}
	q=read();
	while(q--){
		x=read(),y=read(),d=read();
		print(f[x][y][d]);puts("");
	}return 0;
}