蓝桥杯——小明的彩灯（差分数组）-CSDN博客

本文链接：https://blog.csdn.net/m0_65639340/article/details/128646673

文章介绍了如何使用差分数组来解决一个关于彩灯亮度变化的问题。在初始状态下，小明有N个亮度为ai的彩灯，会经历Q次操作，每次操作对一段区间的彩灯亮度加x。文章提供了一个C++程序实现，通过构建差分数组来高效处理这些操作，并确保当亮度为负时输出0。

摘要由CSDN通过智能技术生成

文章目录

差分数组介绍
例题

差分数组介绍

例题

题目描述

小明拥有 N 个彩灯，第 i 个彩灯的初始亮度为a_i
小明将进行 Q 次操作，每次操作可选择一段区间，并使区间内彩灯的亮度 +x（x 可能为负数）。
求 Q次操作后每个彩灯的亮度（若彩灯亮度为负数则输出 0)。

输入描述

第一行包含两个正整数 N，Q，分别表示彩灯的数量和操作的次数。
第二行包含 NN 个整数，表示彩灯的初始亮度。
接下来 QQ 行每行包含一个操作，格式如下：
- l r x，表示将区间l~r的彩灯的亮度 +x。

输出描述

输出共 11 行，包含 NN 个整数，表示每个彩灯的亮度。

输入输出样例

在这里插入图片描述

题解

#include <iostream>
using namespace std;
int main()
{
  // 请在此输入您的代码
  long long N,Q;//数量，操作次数
  long long a,b,c;
  //arr1--原数组
  //arr2--差分数组
  long long arr1[500000],arr2[500000];
  cin>>N>>Q;
  //获取原数组
  for(int i=1;i<=N;i++){
    cin>>arr1[i];
  }
  //得到差分数组
  for(int i=1;i<=N;i++){
    arr2[i]=arr1[i]-arr1[i-1];
  }
  //基于输入改变差分数组
  for(int i=1;i<=Q;i++){
    cin>>a>>b>>c;
    arr2[a] += c;
    arr2[b+1] -= c; 
  }
  //借助差分数组求改后的原数组
  for(int i=1;i<=N;i++){
    arr1[i]=arr2[i]+arr1[i-1]; 
  }
  for(int i=1;i<=N;i++){
    if(arr1[i]<0){
      arr1[i]=0;
    }
  }
  for(int i=1;i<=N;i++){
    cout<<arr1[i]<<" ";
  }
  return 0;
}