CF294A - Shaass and Oskols
题目
Shaass and Oskols - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
思路
设初始电线上鸟数量为arr[n],ai为第i跟电线上鸟数量。
当击中一只鸟,设该鸟为电线序号x上的第y只,左边鸟数量为y - 1,右边为arr[x] - y。
我们只需要维护当前鸟左右电线,x - 1 和 x + 1 号电线,
并且只有当左右电线存在时,其他鸟才会飞到左右相邻电线上,反之飞走。
- 当 x - 1 >= 1,arr[x - 1] += y - 1
- 当 x + 1 <= n,arr[x + 1] += arr[x] - y
- 最后x号电线上鸟数量为0,arr[x] = 0。
代码
#include <bits/stdc++.h>
using namespace std;
// 定义所需变量
int n, m;
int arr[107];
int main() {
cin >> n;
// 数据读入
for (int i = 1; i <= n; i++) {
cin >> arr[i];
}
cin >> m;
// m次击中鸟
while (m--) {
int x, y;
cin >> x >> y;
// x左边有电线
if (x - 1 >= 1) {
// 左边 y - 1 只鸟飞到左边电线
arr[x - 1] += y - 1;
}
// x右边有电线
if (x + 1 <= n) {
// 右边arr[x] - y只鸟飞到右边电线
arr[x + 1] += arr[x] - y;
}
// 将x号电线上鸟数量置为0
arr[x] = 0;
}
// 输出答案
for (int i = 1; i <= n; i++) {
cout << arr[i] << endl;
}
return 0;
}
复杂度分析
- 时间复杂度O(max{m, n})
- 空间复杂度O(n)