AcWing 3729. 改变数组元素

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

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

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

从数组 V尾部插入整数 0。

   2.将位于数组 V末尾的 ai 个元素都变为 1（已经是 1的不予理会）。

注意：

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

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

输入格式

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

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

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

输出格式

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

数据范围

1≤T≤20000,
1≤n≤2×1e5,
0≤ai≤n,
保证一个测试点内所有 n 的和不超过 2×1e5。

输入样例：

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

代码实现：

#include<iostream>

#include<cstring>

using namespace std;

#define int long long

const int N=2e5+5;

int a[N],b[N];

signed main(){

    int t;

    cin>>t;

    while(t--){

        int n;

        cin>>n;

        memset(b,0,sizeof b);

        for(int i=1;i<=n;i++){

            int l=1,r=i;

            scanf("%lld",&a[i]);

            if(a[i]!=0){

                l=max(l,i-a[i]+1);

                b[l]+=1;

                b[r+1]-=1;

            }

        }

        for(int i=1;i<=n;i++){

            b[i]+=b[i-1];

            if(b[i]>=1)cout<<"1 ";

            else cout<<"0 ";

        }

        cout<<endl;

    }

    return 0;

}

AcWing 3729. 改变数组元素的相关教程结束。