蓝桥杯C++大学B组一个月冲刺记录2024/3/25
规则:每日三题
疲倦,树状数组的数学推导证明好麻烦
1.数星星
天空中有一些星星,这些星星都在不同的位置,每个星星有个坐标。
如果一个星星的左下方(包含正左和正下)有 k颗星星,就说这颗星星是 k 级的。
给定星星的位置,输出各级星星的数目。
换句话说,给定 N 个点,定义每个点的等级是在该点左下方(含正左、正下)的点的数目,试统计每个等级有多少个点。
树状数组
当某个星星放置后,会让其右上方的所有星星的等级+1.且根据题目条件:y递增输入(y相同时候x递增输入)
且访问到某个星星后,一定会使某一个等级的星星数量加1
#include<iostream>
using namespace std;
const int N = 1e5;
int tr[N],ans[N];
int n;
int lowbit(int x)
{
return x&-x;
}
void add(int p,int x)
{
for(int i = p;i <= N; i += lowbit(i)) tr[i] += x;//之后访问到的所有星星都要加1
}
int query(int p)
{
int res = 0;
for(int i = p; i; i -= lowbit(i)) res += tr[i];
return res;
}
int main(){
cin >> n;
for(int i = 1;i <= n; ++i)
{
int x,y;
cin >> x >> y;
add(x + 1,1);
ans[query(x + 1)] ++;
}
for(int i = 1;i <= n; ++i) cout << ans[i] << endl;
return 0;
}
2.动态求连续区间和
给定 n 个数组成的一个数列,规定有两种操作,一是修改某个元素,二是求子数列 [a,b]
的连续和。
树状数组 + 模板题
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int p[N];
int str[N];
int n,m;
int lowbit(int x){
return x & -x;
}
void add(int p,int x)
{
for(int i = p;i <= n; i += lowbit(i)) str[i] += x;
return;
}
int query(int p){
int res = 0;
for(int i = p;i;i-=lowbit(i)) res += str[i];
return res;
}
int main()
{
cin >> n >> m;
for(int i = 1;i<=n;++i) cin >> p[i];
for(int j = 1;j <= n; ++j) add(j,p[j]);
while(m--){
int k,x,y;
cin >> k >> x >> y;
if(k == 1) add(x,y);
else{
cout << query(y) - query(x - 1) << endl;
}
}
return 0;
}
3.一个简单的整数问题
给定长度为 N 的数列 A,然后输入 M 行操作指令。
第一类指令形如 C l r d,表示把数列中第 l∼r 个数都加 d。
第二类指令形如 Q x,表示询问数列中第 x 个数的值。
对于每个询问,输出一个整数表示答案。
树状数组 + 差分
在树状数组里面放的是数据差分的值即可
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int p[N];
int str[N];
int n,m;
int lowbit(int x)
{
return x&-x;
}
int add(int p,int x)
{
for(int i = p;i <= n; i+=lowbit(i)) str[i] += x;
}
int query(int p)
{
int res = 0;
for(int i = p; i; i-=lowbit(i)) res += str[i];
return res;
}
int main()
{
cin >> n >> m;
for(int i = 1;i <= n;++i) cin >> p[i];
for(int j = 1;j <= n; ++j) add(j,p[j] - p[j - 1]);
while(m--){
char c;
int l,r,x;
cin >> c;
if(c == 'Q')
{
cin >> x;
cout << query(x) << endl;
}
else
{
cin >> l >> r >> x;
add(l,x);
add(r + 1,-x);
}
}
return 0;
}
4.一个简单的整数问题2
给定一个长度为 N 的数列 A,以及 M 条指令,每条指令可能是以下两种之一:
C l r d,区间内[ l , r ] 所以数全部加d
Q l r,表示询问数列中第 l∼r个数的和。
对于每个询问,输出一个整数表示答案。
树状数组 + 差分 + 数学
这个st2的设定不可谓不妙啊
将求区间和的操作变成了两个在差分数组求前缀和的操作
#include<iostream>
using namespace std;
const int N = 1e5;
typedef long long LL;
int n,m;
int p[N];
LL tr1[N],tr2[N];
int lowbit(int x)
{
return x&-x;
}
void add(LL tr[],int p,LL x)
{
for(int i = p;i<=n;i += lowbit(i)) tr[i] += x;
}
LL query(LL tr[],int p)
{
LL res = 0;
for(int i = p; i; i -= lowbit(i)) res += tr[i];
return res;
}
LL sum(int x)
{
LL ans = query(tr1,x) * (x + 1) - query(tr2,x);
return ans;
}
int main()
{
cin >> n >> m;
for(int i = 1;i <= n; ++i) cin >> p[i];
for(int j = 1;j <= n; ++j)
{
LL t = p[j] - p[j - 1];
add(tr1, j, t);
add(tr2, j, t * j);
}
while(m--){
char c;
LL x,y,z;
cin >> c;
if(c == 'Q')
{
cin >> x >> y;
cout << sum(y) - sum(x - 1) << endl;
}
else
{
cin >> x >> y >> z;
add(tr1,x,z);
add(tr1,y + 1,-z);
add(tr2,x,x*z);
add(tr2,y + 1,(y + 1) * -z);
}
}
return 0;
}