AcWing 3729. 改变数组元素

Posted 2023-03-30 hxss

给定一个空数组 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，数组内元素之间用空格隔开。

数据范围

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;