AcWing 3729. 改变数组元素

发布时间 2023-03-30 14:55:43作者: 回忆、少年

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

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

i次操作的具体流程如下:

  1. 从数组 V尾部插入整数 0

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

注意:

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

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

输入格式

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

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

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

输出格式

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

数据范围

1T20000,
1n2×1e5,
0ain,
保证一个测试点内所有 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;
}