关于vector迭代器的问题

写了个树状dp的题

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

Ural大学有N名职员，编号为1~N。
他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。
每个职员有一个快乐指数，用整数 HiHi 给出，其中 1≤i≤N,1≤i≤N1\leq i\leq N,1\leq i\leq N1≤i≤N,1≤i≤N。
现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。
在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入描述:

第一行一个整数N。
接下来N行，第 i 行表示 i 号职员的快乐指数Hi。
接下来N-1行，每行输入一对整数L, K,表示K是L的直接上司。
最后一行输入0,0。

输出描述:

输出最大的快乐指数。

#include <bits/stdc++.h>
#define ll long long
#define int long long
#define ull unsigned long long
#define vb v.begin()
#define ve v.end()
#define vc v.clear()
#define vs (int)v.size()
#define ss (int)s.size()
#define rs(i) resize(i)
#define ft first
#define sd second
#define inf 0x3f3f3f3f
#define P pair<int, int>
#define Pll pair<ll, ll>
#define Pis pair<int, string>
#define Psi pair<string, int>
#define Pic pair<int, char>
#define Pci pair<char, int>
#define forn(i, a, b) for (int i = a; i < b; i++)
#define fore(i, a, b) for (int i = a; i <= b; i++)
#define rofn(i, a, b) for (int i = a; i > b; i--)
#define rofe(i, a, b) for (int i = a; i >= b; i--)
#define Ios ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
int h[7010];
vector<int> son[7010];
int dp[8010][2] = {0};
void dfs(int x, int fa)
{
	dp[x][1] = h[x];
	dp[x][0] = 0;
	//cout << x << '*' << dp[x][0] << '&' << dp[x][1] << endl;
	fore(i, 0, son[x].size() - 1)//************
	{   
		int j = son[x][i];
		if (j == fa) continue;
		dfs(j, x);
		dp[x][0] += max(dp[j][1], dp[j][0]);
		dp[x][1] += dp[j][0];
	}
	return;
}
signed main()
{
	Ios;
	int n;
	cin >> n;
	fore(i, 1, n) cin >> h[i];
	int sum = 0;
	fore(i, 1, n - 1)
	{
		int l, k;
		cin >> l >> k;
		sum += k;
		son[k].push_back(l);
		son[l].push_back(k);
	}
	int p, q;
	cin >> p >> q;
	dfs(1, -1);
	cout << max(dp[1][0], dp[1][1]) << endl;
	return 0;
}

#include <bits/stdc++.h>
#define ll long long
#define int long long
#define ull unsigned long long
#define vb v.begin()
#define ve v.end()
#define vc v.clear()
#define vs (int)v.size()
#define ss (int)s.size()
#define rs(i) resize(i)
#define ft first
#define sd second
#define inf 0x3f3f3f3f
#define P pair<int, int>
#define Pll pair<ll, ll>
#define Pis pair<int, string>
#define Psi pair<string, int>
#define Pic pair<int, char>
#define Pci pair<char, int>
#define forn(i, a, b) for (int i = a; i < b; i++)
#define fore(i, a, b) for (int i = a; i <= b; i++)
#define rofn(i, a, b) for (int i = a; i > b; i--)
#define rofe(i, a, b) for (int i = a; i >= b; i--)
#define Ios ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
int h[7010];
vector<int> son[7010];
int dp[8010][2] = { 0 };
void dfs(int x, int fa)
{
	dp[x][1] = h[x];
	dp[x][0] = 0;
	//cout << x << '*' << dp[x][0] << '&' << dp[x][1] << endl;
	for(auto j:son[x])//*****************
	{
		if (j == fa) continue;
		dfs(j, x);
		dp[x][0] += max(dp[j][1], dp[j][0]);
		dp[x][1] += dp[j][0];
	}
	return;
}
signed main()
{
	Ios;
	int n;
	cin >> n;
	fore(i, 1, n) cin >> h[i];
	int sum = 0;
	fore(i, 1, n - 1)
	{
		int l, k;
		cin >> l >> k;
		sum += k;
		son[k].push_back(l);
		son[l].push_back(k);
	}
	int p, q;
	cin >> p >> q;
	dfs(1, -1);
	cout << max(dp[1][0], dp[1][1]) << endl;
	return 0;
}

第一个dfs 里用int这么写有一个样例会段错误，第二个用auto的话就可以过了