程序设计天梯赛L3-25

题目
题意: 给定n个点m条有向边，求从S到T有多少条不同的路径。并且判断是否以S为起点的路径均以T为结尾。
思路: 刚开始以为是topsort，但是不对。因为会有其他点乱指，所以得dfs，记忆化搜索一下即可。
时间复杂度: O(n)
代码:

#include<bits/stdc++.h>
#define mem(a,x) memset(a,x,sizeof(a))
using namespace std;
const int N = 502;
int n,m,k,T;
int in[N];
int out[N];
int st,ed;
bool vis[N];
vector<int> va[N];
int ans = 0;
bool flag = true;
int f[N]; //记忆化搜索
int dfs(int cur)
{
	if(f[cur] != -1) return f[cur];
	if(cur == ed) return f[cur] = 1;
	if(!va[cur].size())
	{
		flag = false;
	}
	int res = 0;
	for(int i=0;i<va[cur].size();++i)
	{
		int j = va[cur][i];
		res += dfs(j);
	}
	return f[cur] = res;
}
void solve()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	mem(f,-1);
	for(int i=0;i<m;++i)
	{
		int x,y; cin>>x>>y;
		out[x]++;
		in[y]++;
		va[x].push_back(y);
	}
	cin>>st>>ed;
	ans = dfs(st);
	cout<<ans<<" ";
	if(flag) cout<<"Yes";
	else cout<<"No";
}
signed main(void)
{
	solve();
	return 0;
}