2020牛客寒假算法基础训练营day04 H-坐火车
思路
用树状数组维护\([L,R]\)区间的答案，每次回答完后\(O(logn)\)更新。
具体怎么做呢？
我们假设说目前考虑颜色\(col\)，我左边有\(x\)个\(col\)，右边有\(num-x\)个\(col\)。
那么\(col\)给答案的贡献就是\(x*(num-x)\)。
假设我当前位置的颜色是\(c\)，那么我考虑完我这个位置的时候，我往右边走一步，那么左边多一个\(c\)，右边少一个\(c\)。
所以我只需要用树状数组维护左右每个颜色的乘积，每次修改两次（减去上一次的答案，添加新的答案）。
但是维护乘法爆\(long\ long\)了，我不知道为啥，我觉得按道理来说\(5e5*5e5*5e5=1e17\)不会炸。
如果有明白的朋友请联系我。
接下来我们看看怎么维护这样一个乘积。
假设说我在位置\(i\)，颜色\(c\)左边有\(x\)个，右边有\(sum-x\)个。
此时的答案就是\(x*(sum-x)=x*sum-x^2\)。
继续遍历，到达位置\(j\)，左边的颜色\(c\)有\(x+1\)个，右边有\(sum-x-1\)个。
此时的答案就是\((x+1)*(sum-x-1)=x*sum-x^2-x+sum-x-1\)。
所以我们只需要减去上次的\(x\)，在加上一个\(sum-x-1\)就可以用加法代替乘法更新答案了。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e5+10;
int n, col[maxn], r[maxn], l[maxn];
inline int lowbit(int x){return x&(-x);}
int mxx;
ll c[maxn];
inline void add(int x, ll y){
for(; x <= mxx; x += lowbit(x)) c[x] += y;
}
ll ask(int x){
ll ans = 0;
for(; x; x -= lowbit(x)) ans += c[x];
return ans;
}
ll vis[maxn];
ll used[maxn];
int main()
{
scanf("%d", &n);
for(int i = 1; i <= n; i++)
{
scanf("%d%d%d", &col[i], &l[i], &r[i]);
mxx = max(mxx, max(col[i], max(l[i], r[i])));
}
for(int i = 1; i <= n; i++) vis[col[i]]++;
ll ans = 0;
for(int i = 1; i <= n; i++)
{
int x = col[i];
add(x, -used[x]);
ans = ask(r[i]) - ask(l[i]-1);
used[x]++;
add(x, vis[x]-used[x]);
printf("%lld ", ans);
}
return 0;
}