CF1791F Range Update Point Query 题解
怎么全是线段树啊,来点 DSU。
维护 $f_i$ 表示 $i$ 后第一个 $\ge 10$ 的位置(需要操作的位置),初始 $f_i=i$。
区间修改 $[l,r]$ 时,初始 $i=\operatorname{find}(l)$,修改 $a_i$ 后,若 $a_i<10$,令 $f_i=i+1$。
然后令 $i=\operatorname{find}(i+1)$(跳到下一个需要操作的位置)。
#include <cstdio>
int T, n, m, a[200050], f[200050];
int F(int x) { return x == f[x] ? x : f[x] = F(f[x]); }
int main()
{
scanf("%d", &T);
while (T--)
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i)
scanf("%d", a + i), f[i] = i;
f[n + 1] = n + 1;
for (int i = 0, o, x, y; i < m; ++i)
{
scanf("%d%d", &o, &x);
if (o & 1)
{
scanf("%d", &y);
for (int j = F(x); j <= y; j = F(j + 1))
{
o = a[j];
a[j] = 0;
while (o)
a[j] += o % 10, o /= 10;
if (a[j] < 10)
f[j] = j + 1;
}
}
else
printf("%d\n", a[x]);
}
}
return 0;
}