Acwing-3729. 改变数组元素

给定一个空数组 V 和一个整数数组 a1,a2,…,an.现在要对数组 V 进行 n次操作。第 i次操作的具体流程如下:从数组 V尾部插入整数 0。将位于数组 V末尾的 ai个元素都变为 1(已经是 1的不予理会)。
注意:ai可能为 0,不做任何改变。ai可能大于目前数组 V所包含的元素个数,此时视为将数组内所有元素变为 1。请你输出所有操作完成后的数组 V。
输入格式,第一行包含整数 T,表示共有 T 组测试数据。每组数据第一行包含整数 n。第二行包含n个整数 a1,a2,…,an。
输出格式,每组数据输出一行结果,表示所有操作完成后的数组 V,数组内元素之间用空格隔开。
数据范围 1≤T≤20000 1≤n≤2×105 0≤ai≤n 保证一个测试点内所有 n 的和不超过 2×105
输入样例: 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>
#include <algorithm>
using namespace std;

const int N=2e5+10;
int g[N];

int main()
{
          
   
    int t,n;
    cin>>t;
    while(t--)
    {
          
   
        cin>>n;
        memset(g,0,(n+1)*4);
        for(int i=1; i<=n; i++)
        {
          
   
            int x;
            cin>>x;
            x=min(i,x);
            int l=i-x+1,r=i;
            g[l]++;g[r+1]--;//差分核心代码
        }
        for(int i=1; i<=n; i++) g[i]+=g[i-1];//差分数组前缀和
        for(int i=1; i<=n; i++) cout<<!!g[i]<< ;//!!g[i],是0输出0,非0输出1
        cout<<
;
    }
    return 0;
}
经验分享 程序员 微信小程序 职场和发展