问题描述
小葱喜欢除法,所以他给了你N个数a1, a2, ⋯, aN,并且希望你执行M次操作,每次操作可能有以下两种:
给你三个数l, r, v,你需要将al, al+1, ⋯, ar之间所有v的倍数除以v。
给你两个数l, r,你需要回答al + al+1 + ⋯ + ar的值是多少。
输入格式
第一行两个整数N, M,代表数的个数和操作的次数。接下来一行N个整数,代表N个数一开始的值。
接下来M行,每行代表依次操作。每行开始有一个整数opt。如果opt=1,那么接下来有三个数l, r, v,代表你需要回答第l个数到第r个数的和。
输出格式
对于每一次的第二种操作,输出一行代表这次操作所询问的值。
样例输入
5 3
1 2 3 4 5
2 1 5
1 1 3 2
2 1 5
样例输出
15
14
评测用例规模与约定
对于30%的评测用例,1 ≤ N, M ≤ 1000;
对于另外20%的评测用例,第一种操作中一定有l = r;
对于另外20%的评测用例,第一种操作中一定有l = 1 , r = N;
对于100%的评测用例,1 ≤ N, M ≤ 105,0 ≤ a1, a2, ⋯, aN ≤ 106, 1 ≤ v ≤ 106, 1 ≤ l ≤ r ≤ N。
解题思路
本题只拿了50分,其余五组数据均超时,如果有谁想到了优化方法,欢迎在底下评论哟(^U^)ノ~YO。
非常感谢在底部评论的@_zlWang大佬,利用树状数组和一些小优化完美的AC了本道题。
#include <iostream>
using namespace std;
int a[100001];
int main()
{
std::ios::sync_with_stdio(false);
int n, m, l, r, v, flag;
cin >> n >> m;
for (int i = 0; i < n; i++)
{
cin >> a[i];
}
while (m--)
{
cin >> flag;
if (flag == 1)
{
cin >> l >> r >> v;
for (int i = l - 1; i < r; i++)
{
if (!(a[i] % v))
{
a[i] /= v;
}
}
}
else if (flag == 2)
{
cin >> l >> r;
long long sum = 0;
for (int i = l - 1; i < r; i++)
{
sum += a[i];
}
cout << sum << endl;
}
}
return 0;
}
修改优化后的代码,貌似已经被改的面目全非了(⊙o⊙)…
#include <iostream>
using namespace std;
int n, m, l, r, v, flag, a[100001];
long long t[100001];//定义树状数组
int lowbit(int x)//清除x的二进制高位1
{
return x & (-x);
}
void change(int x, int p)//将第x个数加p
{
while (x <= n)
{
t[x] += p;
x += lowbit(x);
}
}
long long sum(int k)//前k个数的和
{
long long ans = 0;
while (k > 0)
{
ans += t[k];
k -= lowbit(k);
}
return ans;
}
long long ask(int l, int r)//求l-r区间和
{
return sum(r) - sum(l - 1);
}
int main()
{
std::ios::sync_with_stdio(false);
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
cin >> a[i];
change(i, a[i]);
}
while (m--)
{
cin >> flag;
if (flag == 1)
{
cin >> l >> r >> v;
if (v == 1) continue;
for (int i = l; i <= r; i++)
{
if (a[i] >= v && a[i] % v == 0)
{
change(i, a[i] / v - a[i]);
a[i] /= v;
}
}
}
else if (flag == 2)
{
cin >> l >> r;
cout << ask(l, r) << endl;
}
}
return 0;
}