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;
}
posted @ 2023-02-07 21:04  5k_sync_closer  阅读(69)  评论(0编辑  收藏  举报  来源