【蓝桥杯集训】Acwing3729. 改变数组元素

题目

给定一个空数组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×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

 分析题目

由这题不难想到,设整数数组为arr,若arr[i]为m (0<m) ,则V[i]到V[i-m+1]都是1(若m>i+1,则是V[i]到V[0]为1。思路十分简单,但是如何批量让数组某一段为1,且时间复杂度最小呢?

这里就涉及到差分,详情请看知识点,然后我们上代码

#include<bits/stdc++.h>

using namespace std;

const int N=200010;

int main() {
    int  n;
    cin >> n;
    int x;
    int v[N];
    while(n--) {
        int num;
        cin >> num;
        memset(v,0,(num+1)*4);
        for (int j = 1; j<=num; j++) {
            cin >> x;
            if (x) {
                int p = min(j+1,x);
                v[j-p+1]++;v[j+1]--;
            }
        }

        for (int j = 1; j <= num; j++) {
            v[j]+=v[j-1];
        }

        for(int j=1;j<=num;j++){
            printf("%d ",!!v[j]);
        }
        puts("");
    }
    return 0;
}

注意事项

V数组里只有0和1,如果使用差分会出现超过1的情况,所以需要使用!!V[i]

知识点:

差分:

如果一个数组arr1的大小为n,想要对其下标a~b进行批量加k的操作。可以先得到它的差分数组arr2

arr2[i]=arr1[i]-arr1[i-1]

再对a~b进行批量操作

arr2[a]+=k,arr2[b+1]-=k

最后再利用前缀和求出新的arr1

arr1[i]=arr1[i-1]+arr2[i]

memset:

按照字节对内存块进行初始化,不能用它将int数组出初始化为0和-1之外的其他值

void *memset(void *s, int c, size_t n); 

memset(v,0,(num+1)*4)

s代表填充的内存块,c为填充值,n表示填充的个数(本题实例中(num+1)*4代表填充num+1,乘4则是因为int的4个字节)
 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值