动态规划+vector解旅行计划问题

动态规划+vector解旅行计划问题

题目描述

小明要去一个国家旅游。这个国家有#N个城市，编号为1至N，并且有M条道路连接着，小明准备从其中一个城市出发，并只往东走到城市i停止。

所以他就需要选择最先到达的城市，并制定一条路线以城市i为终点，使得线路上除了第一个城市，每个城市都在路线前一个城市东面，并且满足这个前提下还希望游览的城市尽量多。

现在，你只知道每一条道路所连接的两个城市的相对位置关系，但并不知道所有城市具体的位置。现在对于所有的i，都需要你为小明制定一条路线，并求出以城市ii为终点最多能够游览多少个城市。

输入格式

第1行为两个正整数N, M。

接下来M行，每行两个正整数x, y，表示了有一条连接城市x与城市y的道路，保证了城市x在城市y西面。

输出格式

N行，第i行包含一个正整数，表示以第i个城市为终点最多能游览多少个城市。

输入输出样例

输入 #1

5 6
1 2
1 3
2 3
2 4
3 4
2 5

输出 #1

1
2
3
4
3

说明/提示

均选择从城市1出发可以得到以上答案。

对于20%的数据，N ≤ 100；

对于60%的数据，N ≤ 1000；

对于100%的数据，N ≤ 100000,M ≤ 200000。

解题思路

用向量v[i]存储所有可以游览城市i的城市，那么v[i][j]则表示这些城市中的第 j个

f[i]表示以第 i个城市为终点最多能游览的城市数

那么很容易想到，要求以城市i为终点的最大游览数量，就要求以能游览城市i的城市为终点的最大游览数量,所以状态转移方程就是f[now]=max(f[now],dp(v[now][i])+1)

最后只要循环输出dp[i]即可

完整代码

#include <iostream>
#include <vector>
#include <string.h>
using namespace std; 
int f[100000];//f[i]表示以第 i个城市为终点最多能游览的城市数
vector<int> v[100000];//开很多很多向量存储位置关系，v[i]存储所有能游览第 i个城市的城市

int dp(int now)//当前城市
  {
  	if(f[now]!=-1) return f[now];//若不是初值的 -1，说明该城市已经计算过了 
  	f[now]=1;//至少可以游览自己 
  	for(int i=0;i<v[now].size();i++)//遍历 v[now]向量存储所有能游览城市 now的城市
  	  {
  	    f[now]=max(f[now],dp(v[now][i])+1);//v[now][i]则表示这些城市中的第 i个	
	  }
  	return f[now]; 
  	
  }

int main()
  {
  	memset(f,-1,sizeof(f));//赋初值为 -1 
  	int n,m;
  	cin>>n>>m;
  	for(int i=1;i<=m;i++)
  	  {
  	  	int x,y;
  	  	cin>>x>>y;
  	  	v[y].push_back(x);//把能游览 y城市的城市都存在 v[y]里 
	  }
	for(int i=1;i<=n;i++)
	  {
	  	cout<<dp(i)<<endl;
	  }
  	return 0;
  }