F. Range Update Point Query

题意

题意：给我们n个数，m次操作，每次操作一种是修改，一种是查询。
修改的话是把l，r之间的每一个数变成本身的每一位数的和，查询操作是求下标是x的数是多少。

思路

思路：一开始使用线段树做的，但是每次修改加查询的话会T。

我们考虑用set。我们可以用set中自带的lower_bound函数，帮助我们修改l-r之间的数。

#include <bits/stdc++.h>
using namespace std;

const int N = 2e5 + 10;
int a[N];
set<int> se;

int f(int x)
{
    int num = 0;
    while (x)
    {
        int g = x % 10;
        num += g;
        x /= 10;
    }
    return num;
}

void solve()
{
    se.clear();
    int n, q;
    cin >> n >> q;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        if (a[i] != f(a[i]))
            se.insert(i);
    }
    se.insert(n + 1);//保证set的lower_bound边界
    while (q--)
    {
        int op;
        cin >> op;
        if (op == 1)
        {
            int l, r;
            cin >> l >> r;
            while (true)
            {
                int pos = (*se.lower_bound(l));//大于等于当前左边界的第一个数是谁
                if (pos > r)//如果左边界大于右边界的话就break
                    break;
                a[pos] = f(a[pos]);
                if (a[pos] == f(a[pos]))
                    se.erase(pos);
                l = pos + 1;
            }
        }
        else
        {
            int x;
            cin >> x;
            cout << a[x] << "\n";
        }
    }
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    int T;
    cin >> T;
    while (T--)
    {
        solve();
    }
    return 0;
}