|BZOJ 3211|树状数组|并查集|花神游历各国

最新推荐文章于 2019-12-09 15:12:59 发布

NotFound1

最新推荐文章于 2019-12-09 15:12:59 发布

阅读量597

点赞数

分类专栏： BZOJ 树 - 树状数组树 - 并查集

本文链接：https://blog.youkuaiyun.com/Darost/article/details/57084963

版权

BZOJ 同时被 3 个专栏收录

33 篇文章

订阅专栏

树 - 并查集

8 篇文章

订阅专栏

树 - 树状数组

6 篇文章

订阅专栏

本文介绍了解决BZOJ 3211问题的方法，使用树状数组结合并查集来优化操作，针对数值进行平方根运算时可能导致的速度下降问题，提出了一种高效的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

BZOJ传送门
树状数组裸题，但是因为sqrt的速度很慢，所以我们在一个数为1或者为0就删掉它(用并查集维护)
ps:交题时注意文件声明部分要注释掉！坑死了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<iostream>
#define ms(i,j) memset(i, j, sizeof i);
#define ll long long
using namespace std;
const int MAXN = 100000 + 5;
ll a[MAXN], f[MAXN];
int data[MAXN];
int n,m;
int lowbit(ll x) {return x&(-x);}
int find(int x)
{
    if (f[x]==x) return x;
    else return f[x] = find(f[x]);
}
ll sum(int x)
{
    ll ret = 0;
    for (ll i=x;i>0;i-=lowbit(i))
    {
        ret += (ll)a[i];
    }
    return ret;
}
int update(int x, int c)
{
    for (ll i=x;i<=n;i+=lowbit(i))
    {
        a[i] += c;
    }
}
int main()
{
    //freopen("bzoj3211.in", "r", stdin); freopen("bzoj3211.out", "w", stdout);
    scanf("%d\n", &n); ms(a, 0);
    for (int i=1;i<=n+1;i++) f[i] = i;
    for (int i=1;i<=n;i++) 
    {
        scanf("%lld", &data[i]);
        if (data[i]==1||data[i]==0) f[i] = find(i+1);
        update(i, data[i]); 
    }
    scanf("%d", &m);
    for (int i=1;i<=m;i++)
    {
        int x, l, r;
        scanf("%d%d%d", &x, &l, &r);
        if (x==1) printf("%lld\n", sum(r)-sum(l-1)); else if (x==2)
        {
            for (int j=find(l);j<=r;j=find(j+1))
            {
                update(j, (ll)(sqrt(data[j]))-data[j]);
                data[j] = (ll)(sqrt(data[j]));
                if(data[j]==1)
                {
                    f[j] = find(j+1);
                }
            }
        }
    }
    return 0;
}