前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >改变数组元素

改变数组元素

作者头像
浪漫主义狗
发布2023-02-21 11:52:14
5830
发布2023-02-21 11:52:14
举报
文章被收录于专栏:HAUE_LYS'BlogHAUE_LYS'Blog

Original link

思想

  • 前缀和与差分。
  • 对于 a[i] 的操作,构建差分数组 b[i] 记录被操作的区间。
  • 利用前缀和可知所有被操作的区间,即 b[i] != 0 的区间。

代码

代码语言:javascript
复制
#include <bits/stdc++.h>
using namespace std;

const int N = 1e6 + 3;

int a[N], b[N];

void solve(){
    int n; cin >> n; 
    for(int i = 1; i <= n; i ++){
        b[i] = 0;  //重置
        cin >> a[i];  //记录操作
    }
    for(int i = 1; i <= n; i ++){
        if(a[i] >= i) b[1] ++, b[i + 1] --;  //a[i] 超出范围,则当前全部区间都被操作了
        else if(a[i] < i && a[i] != 0) b[i - a[i] + 1] ++, b[i + 1] --;  //否则操作后 a[i] 个数,正数为第 i - a[i] + 1 的位置开始
    }
    for(int i = 1; i <= n; i ++){
        b[i] += b[i - 1];  //前缀和计算每个区间被操作的次数
        if(b[i] != 0) cout << 1 << ' ';  //不为 0 说明该位置被操作过
        else cout << 0 << ' ';
    }
    cout << endl;
}

int main(){
    int _; cin >> _;
    while(_ --) solve();
    return 0;
}
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2023-2-14 1,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档