每日打卡day6

៚ོ朝夕ོོ ͜✿ ҉҉҉҉҉2023-07-18 23:20

题目如下：

给定一个空数组 V 和一个整数数组 a1,a2,...,an。

现在要对数组 V 进行 n 次操作。

第 i 次操作的具体流程如下：

从数组 V 尾部插入整数 0。
将位于数组 V 末尾的 ai 个元素都变为 1（已经是 1 的不予理会）。

注意：

ai 可能为 0，即不做任何改变。
ai 可能大于目前数组 V 所包含的元素个数，此时视为将数组内所有元素变为 11。

请你输出所有操作完成后的数组 V。

输入格式

第一行包含整数 T，表示共有 T 组测试数据。

每组数据第一行包含整数 n。

第二行包含 n 个整数 a1,a2,...,an。

输出格式

每组数据输出一行结果，表示所有操作完成后的数组 V，数组内元素之间用空格隔开。

数据范围

1≤T≤20000,

1≤n≤2×10^5,

0≤ai≤n,

保证一个测试点内所有 n 的和不超过 2×10^5。

输入样例：

复制代码

3
6
0 3 0 0 1 3
10
0 0 0 1 0 5 0 0 0 2
3
0 0 0

输出样例：

复制代码

1 1 0 1 1 1
0 1 1 1 1 1 0 0 1 1
0 0 0

cpp 复制代码

#include <iostream>
#include <cstring>
using namespace std;

int main(){
    int T = 0;
    cin >> T;
    while(T--){
        int n = 0;
        cin >> n;
        int res[n + 1];
        memset(res, 0, sizeof(res));
        for(int i = 1; i <= n; i++){
            int x = 0;
            cin >> x;
            x = min(x, i);
            int l = i - x + 1, r = i + 1;
            res[l]++, res[r]--;
        }
        for(int i = 1; i <= n; i++)  res[i] += res[i-1];
        for(int i = 1; i <= n; i++)  cout << !!res[i] << " ";
        cout << endl;
    }
    return 0;
}

思路：差分数组处理区间修改值问题，题目大意：将V数组（总长为n）的第n - a[i] 到 i 修改成1

初始化V数组，使其长度为n,值为0，利用差分，完成区间修改，前缀和求每个区间增1后的值，最后将修改的值变成1，没修改的值保持不变