写作背景:
由于本人是个菜鸟,所以没写出来这个题目,但是我在看这道题的题解之后自己写了一遍出现了一个bug,一直AC不了,所以分享出来,万一有人跟我一样想的呢?嘿嘿,建议先写一下题目,看一下这个题解再看这个文章。
题目:https://www.luogu.com.cn/problem/P11448
参考的题解:题解:P11448 「ALFR Round 3」D 核裂变 - 洛谷专栏 (luogu.com.cn)
本人写的代码:
#include<iostream>
#include<vector>
#include<utility>
#include<queue>
using namespace std;
using gg = long long;
const gg Mod= 998244353;
const gg N = 5e5 + 5;
gg ans[N];
gg a[N];
vector<gg>e[N];
bool vis[N];
gg n, k;
void BFS(gg s)
{
queue<pair<gg, gg>>q;
q.push({s,0});
while (q.size())
{
gg a1 = q.front().first;
gg b = q.front().second;
q.pop();
if (b >= k)return;
for (gg i : e[a1])
{
if (vis[i])ans[i] = ((ans[i] + (k - b) % Mod)) % Mod;
else {
ans[i] = (ans[i] + (a[i] + 1) * ((k - b) % Mod)%Mod) % Mod;
q.push({ i,++b});
vis[i] = 1;
}
}
}
}
int main()
{
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
cin >> n >> k;
for (gg i = 0; i <= n; i++)
{
cin >> a[i];
for (gg j = 1; j <= a[i]; j++)
{
gg x; cin >> x;
e[i].push_back(x);
}
}
BFS(0);
for (gg i = 1; i <= n; i++)cout << ans[i] << " ";
}
错误的地方就是:
q.push({ i,++b});
原文写的是:
q.push({ i,b+1});
看起来一样是吧?实际上使用原来样例(题目给的)结果还是正确的,但是我们稍微修改一下:看看b是怎么变的
#include<iostream>
#include<vector>
#include<utility>
#include<queue>
using namespace std;
using gg = long long;
const gg Mod= 998244353;
const gg N = 5e5 + 5;
gg ans[N];
gg a[N];
vector<gg>e[N];
bool vis[N];
gg n, k;
void BFS(gg s)
{
queue<pair<gg, gg>>q;
q.push({s,0});
while (q.size())
{
gg a1 = q.front().first;
gg b = q.front().second;
cout << '\n'<<"+++" << b << "+++";
q.pop();
if (b >= k)return;
for (gg i : e[a1])
{
if (vis[i])ans[i] = ((ans[i] + (k - b) % Mod)) % Mod;
else {
ans[i] = (ans[i] + (a[i] + 1) * ((k - b) % Mod)%Mod) % Mod;
q.push({ i,++b});
vis[i] = 1;
}
}
cout << '\n' << "***" << b << "***";
}
}
int main()
{
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
cin >> n >> k;
for (gg i = 0; i <= n; i++)
{
cin >> a[i];
for (gg j = 1; j <= a[i]; j++)
{
gg x; cin >> x;
e[i].push_back(x);
}
}
BFS(0);
for (gg i = 1; i <= n; i++)cout << ans[i] << " ";
}
结果:输入:3 3 1
1
1 2
1 3
1 1这是输出:
+++0+++
***1***
+++1+++
***2***
+++2+++
***3***
+++3+++6 4 2
换成正确的代码试一试:
输入:3 3 1
1
1 2
1 3
1 1输出:+++0+++
***0***
+++1+++
***1***
+++2+++
***2***
+++3+++6 4 2
看起来都差不多是吧,那我们在换一个例子:
错误代码运行结果:
输入:3 3 2
1 3
1 2
2 1 3
2 1 2输出:+++0+++
***2***
+++1+++
***2***
+++2+++
***2***
+++2+++
***2***8 7 7
正确代码运行结果:
输入:3 3 2
1 3
1 2
2 1 3
2 1 2输出:+++0+++
***0***
+++1+++
***1***
+++1+++
***1***
+++2+++
***2***9 8 10
结果不一样了,分析一下:为什么不能用++b来替代b+1?b+1之后b的值不变,这样保证在
这个for循环里面b的值要不变,如果成++b,那么这个循环里面的b的值就不能保持恒定了,这就发生了错误,至于为什么前面的例子是正确的是因为输入的第三个数据是1,这个循环就循环了一次,所以b的值变不变看不出来,最终结果是一样的,但是输入的第三个数据如果不是1,那么我写的那个代码(++b)就错了。
嘿嘿,我也借这个题目好好学一学图。