直接看代码没有找到思维漏洞,然后找了柳诺的代码,修改了测试用例、
7 3
3 2 3 4
0
2 5 6
2 3 1
2 3 4
1 4
1 5
4 1 3 5 7
柳诺的运行结果是:
5
5
5
0
而我是5 6 4 0
#define _CRT_SECURE_NO_WARNINGS 1
#include<bits/stdc++.h>
using namespace std;
vector<int> adj[1010];
bool vis[1010] = {
false };
int p = 0,maxl,n,layer;
void bfs(int s) {
queue<int> q;
q.push(s);
layer=0;
p=0;
while (!q.empty()) {
if (layer == maxl)
return;
int v = q.front();
q.pop();
vis[v] = true;
for (int i = 0; i <adj[v].size(); i++) {
if (vis[adj[v][i]] == false) {
q.push(adj[v][i]);
vis[adj[v][i]] == true;
p++;
}
}
layer++;
}
}
int main() {
int test;
cin >> n >> maxl;
for (int i = 1; i <= n; i++) {
int num, temp;
cin >> num;
for (int j = 0; j < num; j++) {
cin >> temp;
adj[temp].push_back(i);
}
}
cin >> test;
int a;
for (int i = 0; i < test; i++) {
cin >> a;
memset(vis,false,sizeof(vis));
bfs(a);
cout << p << endl;
}
return 0;
}
调试发现是因为layer不对,应该让当前层的所有结点的下一层访问完才能layer++,而不是访问完其中一个节点的下一层就layer++
#define _CRT_SECURE_NO_WARNINGS 1
#include<bits/stdc++.h>
using namespace std;
struct node {
int data, layer = 0;
}user;
vector<node> adj[1010];
bool vis[1010] = {
false };
int p = 0, maxl, n;
void bfs(node s) {
queue<node> q;
q.push(s);
s.layer = 0;
p = 0;
while (!q.empty()) {
node v = q.front();
if (v.layer == maxl)
return;
q.pop();
vis[v.data] = true;
for (int i = 0; i < adj[v.data].size(); i++) {
if (vis[adj[v.data][i].data] == false) {
adj[v.data][i].layer = v.layer + 1;
q.push(adj[v.data][i]);
vis[adj[v.data][i].data] = true;
p++;
}
}
}
}
int main() {
int test;
cin >> n >> maxl;
for (int i = 1; i <= n; i++) {
int num, temp;
cin >> num;
user.data = i;
for (int j = 0; j < num; j++) {
cin >> temp;
adj[temp].push_back(user);
}
}
cin >> test;
int a;
for (int i = 0; i < test; i++) {
cin >> a;
user.data = a;
memset(vis, false, sizeof(vis));
bfs(user);
cout << p << endl;
}
return 0;
}
写之前理清楚什么是节点的值和层数。注意细节,不小心把赋值号写成了==,调试出来了。。