2021-05-18 acwing线段树,分块练习

243. 一个简单的整数问题2

1.题意:
2.题解:

简单个鬼鬼

3.ac代码:
#include<bits/stdc++.h>
#include<math.h>
#include<string>
using namespace std;
const int N=5e5+10;
typedef long long ll;
ll a[N];
struct Node{
    int l,r;
    ll sum,add;
}tr[N<<2];
void pushdown(int u){//下放
    if(tr[u].add){
        tr[u<<1].add+=tr[u].add;
        tr[u<<1].sum+=(ll)(tr[u<<1].r-tr[u<<1].l+1)*tr[u].add;
        tr[u<<1|1].add+=tr[u].add;
        tr[u<<1|1].sum+=(ll)(tr[u<<1|1].r-tr[u<<1|1].l+1)*tr[u].add;
        tr[u].add=0;
    }

}
void pushup(int u){
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;

}
void build(int u,int l,int r){
    if(l==r){
        tr[u]={l,r,a[l],0};
    }else{
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid);
        build(u<<1|1,mid+1,r);
        pushup(u);
    }
}
void change(int u,int l,int r,int x){//区间加值
    if(tr[u].l>=l&&tr[u].r<=r){
        tr[u].sum+=(ll)(tr[u].r-tr[u].l+1)*x;
        tr[u].add+=x;
    }else{
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(l<=mid) change(u<<1,l,r,x);
        if(r>mid) change(u<<1|1,l,r,x);
        pushup(u);
    }
}
ll query(int u,int l,int r){//查找区间和
    if(tr[u].l>=l&&tr[u].r<=r){
        return tr[u].sum;
    }
    pushdown(u);
    int  mid=tr[u].l+tr[u].r>>1;
    ll sum=0;
    if(l<=mid) sum+=query(u<<1,l,r);
    if(r>mid) sum+=query(u<<1|1,l,r);
    return sum;

}
int main(){
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>a[i];
    build(1,1,n);
    string op;
    int l,r,d;
    for(int i=1;i<=m;i++){
        cin>>op>>l>>r;
        if(op=="C"){
            cin>>d;
            change(1,l,r,d);;
        }else{
            cout<<query(1,l,r)<<endl;
        }
    }
}

虽然线段树做一点不简单,但是用分块做很简单!!!,数据量1e5就可用,处理分块的总时间复杂度是0(/n)

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
typedef long long ll;
ll a[N];
ll sum[N];
int st[N],ed[N];
int bel[N];
int sz[N];
int mark[N];
void init(int n){
	int sq=sqrt(n);
	for(int i=1;i<=sq;i++){
		st[i]=n/sq*(i-1)+1;//st[i]表示第i号第一个元素的下标 
		ed[i]=n/sq*i;//ed[i]表示第i号元素最后一个元素的下标 
	}
	ed[sq]=n;//把剩余的一小块纳入最后一块中 
	for(int i=1;i<=sq;i++){
		for(int j=st[i];j<=ed[i];j++){
			bel[j]=i;//第j号元素归属i块 
		}
	}
	//求区间块数 
	for(int i=1;i<=sq;i++){
		sz[i]=ed[i]-st[i]+1;
	}
	//求区间和 
	for(int i=1;i<=sq;i++){
		for(int j=st[i];j<=ed[i];j++){
			sum[i]+=a[j];
		}
	}
//	for(int i=1;i<=sq;i++){
//		printf("----%d %d %d\n",st[i],ed[i],sum[i]);
//	}
}
ll query(int x,int y){
	ll s=0;
	if(bel[x]==bel[y]){
		for(int i=x;i<=y;i++){
			s+=a[i]+mark[bel[i]];
		} 
	}
	else{
		for(int i=x;i<=ed[bel[x]];i++){
			s+=a[i]+mark[bel[i]];
		}
		for(int i=st[bel[y]];i<=y;i++){
			s+=a[i]+mark[bel[i]];
		}
		for(int i=bel[x]+1;i<bel[y];i++){
			s+=sum[i]+mark[i]*sz[i];
		}
	}
	return s;
}
void change(int x,int y,int k){
	if(bel[x]==bel[y]){
		for(int i=x;i<=y;i++){
			a[i]+=k;
			sum[bel[i]]+=k;
		}
	}else{
		for(int i=x;i<=ed[bel[x]];i++){
			a[i]+=k;
			sum[bel[i]]+=k;
		}
		for(int i=st[bel[y]];i<=y;i++){
			a[i]+=k;
			sum[bel[i]]+=k;
		}
		for(int i=bel[x]+1;i<bel[y];i++){
			mark[i]+=k;//对中间块打上标记 
		}
	}
}
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	
	//分块
	 init(n);
	 int sq=sqrt(n);

	int x,y,k;
	string op;
	for(int i=1;i<=m;i++){
		cin>>op>>x>>y;
		if(op=="C"){
			cin>>k;
			change(x,y,k);
		}else{
			cout<<query(x,y)<<endl;
		}
	}
	//区间修改
	 
	
}

洛谷P2801 教主的魔法

1.题意:
2.题解:
3.ac代码:

疯掉!,各种问题,这是90分的还有一个数据T了,我真不知道错哪了

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
typedef long long ll;
int a[N];
ll sum[N];
int st[N],ed[N];
int bel[N];
int size[N];
int mark[N];
vector<int> v[N];
void init(int n){
    int sq=sqrt(n);
    for(int i=1;i<=sq;i++){
        st[i]=n/sq*(i-1)+1;//st[i]表示第i号第一个元素的下标
        ed[i]=n/sq*i;//ed[i]表示第i号元素最后一个元素的下标
    }
    ed[sq]=n;//把剩余的一小块纳入最后一块中
    for(int i=1;i<=sq;i++){
        for(int j=st[i];j<=ed[i];j++){
            bel[j]=i;//第j号元素归属i块
            v[i].push_back(a[j]);
        }
        sort(v[i].begin(),v[i].end());
    }
    //求区间块数
    for(int i=1;i<=sq;i++){
        size[i]=ed[i]-st[i]+1;
    }

    //求区间和
//	for(int i=1;i<=sq;i++){
//		for(int j=st[i];j<=ed[i];j++){
//			sum[i]+=a[j];
//
//		}
//	}
//	for(int i=1;i<=sq;i++){
//		printf("----%d %d %d\n",st[i],ed[i],sum[i]);
//	}
}
void update(int b) // 更新排序后的数组
{
    for (int i = 0; i <=size[b]; ++i)
        v[b][i]=a[st[b]+i];
    sort(v[b].begin(), v[b].end());
}
int query(int x,int y,ll c){
    int s=0;
    if(bel[x]==bel[y]){
        for(int i=x;i<=y;i++){
            if(a[i]+mark[bel[x]]>=c) s++;
        }
    }else{
        for(int i=x;i<=ed[bel[x]];i++){
            if(a[i]+mark[bel[x]]>=c) s++;
        }
        for(int i=st[bel[y]];i<=y;i++){
            if(a[i]+mark[bel[y]]>=c) s++;
        }
        for(int i=bel[x]+1;i<bel[y];i++){
            s+=v[i].end()-lower_bound(v[i].begin(),v[i].end(),c-mark[i]);
        }
    }
    return s;
}
void change(int x,int y,int k){
    if(bel[x]==bel[y]){
        for(int i=x;i<=y;i++){
            a[i]+=k;
            //	sum[bel[i]]+=k;
        }
        update(bel[x]);
    }else{
        for(int i=x;i<=ed[bel[x]];i++){
            a[i]+=k;
            //	sum[bel[i]]+=k;
        }
        update(bel[x]);
        for(int i=st[bel[y]];i<=y;i++){
            a[i]+=k;
            //	sum[bel[i]]+=k;
        }
        update(bel[y]);
        for(int i=bel[x]+1;i<bel[y];i++){
            mark[i]+=k;//对中间块打上标记
        }
    }

}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }

    //分块
    init(n);

    int x,y,k,c;
    char op;
    for(int i=1;i<=m;i++){
        scanf(" %c %d %d %d",&op,&x,&y,&c);
        if(op=='M'){
            change(x,y,c);
        }else{
            printf("%d\n",query(x,y,c));
           // cout<<query(x,y,c)<<endl;
        }
    }
    //区间修改


}

这是满分的

#include<bits/stdc++.h>
using namespace std;
const int N=5e5+10;
typedef long long ll;
const int MAXN = 1000005, SQ = 1005;
int st[SQ], ed[SQ], size[SQ], bel[MAXN];
void init_block(int n) // 初始化
{
    int sq = sqrt(n);
    for (int i = 1; i <= sq; ++i)
    {
        st[i] = n / sq * (i - 1) + 1;
        ed[i] = n / sq * i;
    }
    ed[sq] = n;
    for (int i = 1; i <= sq; ++i)
        for (int j = st[i]; j <= ed[i]; ++j)
            bel[j] = i;
    for (int i = 1; i <= sq; ++i)
        size[i] = ed[i] - st[i] + 1;
}
int A[MAXN], mark[SQ];
vector<int> v[SQ]; // 这里用vector存排序后的数组
void update(int b) // 更新排序后的数组
{
    for (int i = 0; i <= size[b]; ++i)
        v[b][i] = A[st[b] + i];
    sort(v[b].begin(), v[b].end());
}
int main()
{
    int n,m;
    scanf("%d %d",&n,&m);
    int sq = sqrt(n);
    init_block(n);
    for (int i = 1; i <= n; ++i)
        scanf("%d",&A[i]);
    for (int i = 1; i <= sq; ++i)
        for (int j = st[i]; j <= ed[i]; ++j)
            v[i].push_back(A[j]);
    for (int i = 1; i <= sq; ++i)
        sort(v[i].begin(), v[i].end());
    while (m--)
    {
        char o;
        scanf(" %c", &o);
        int l,r,k;
        scanf("%d%d%d",&l,&r,&k);
       // int l = read(), r = read(), k = read();
        if (o == 'M')
        {
            if (bel[l] == bel[r]) // 如果同属一块直接暴力
            {
                for (int i = l; i <= r; ++i)
                    A[i] += k;
                update(bel[l]);
                continue;
            }
            for (int i = l; i <= ed[bel[l]]; ++i) // 对零散块暴力处理
                A[i] += k;
            for (int i = st[bel[r]]; i <= r; ++i)
                A[i] += k;
            update(bel[l]);
            update(bel[r]);
            for (int i = bel[l] + 1; i < bel[r]; ++i) // 打上标记
                mark[i] += k;
        }
        else
        {
            int tot = 0;
            if (bel[l] == bel[r])
            {
                for (int i = l; i <= r; ++i)
                    if (A[i] + mark[bel[l]] >= k)
                        tot++;
                printf("%d\n", tot);
                continue;
            }
            for (int i = l; i <= ed[bel[l]]; ++i)
                if (A[i] + mark[bel[l]] >= k)
                    tot++;
            for (int i = st[bel[r]]; i <= r; ++i)
                if (A[i] + mark[bel[r]] >= k)
                    tot++;
            // 二分查找k-mark[i]的位置,因为整块都加上了mark[i]其实就相当于k减去mark[i]
            for (int i = bel[l] + 1; i < bel[r]; ++i) 
                tot += v[i].end() - lower_bound(v[i].begin(), v[i].end(), k - mark[i]);
            printf("%d\n", tot);
        }
    }
    return 0;
}

SPOJ DQUERY - D-query

1.题意:
2.题解:

依旧是外网题不知道过了没

3.ac代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+10;
int sq;
struct query{
	int l,r,id;
	bool operator<(const query&o) const{
		if(l/sq!=o.l/sq){
			return l<o.l;
		}
		if(l/sq&1){
			return r<o.r;
		}
		return r>o.r;
	}
}Q[N];
int cur;
int a[N],ans[N],cnt[N];
inline void add(int p){
	if(cnt[a[p]]==0){
		cur++;
	}
	cnt[a[p]]++;
}
inline void del(int p){
	cnt[a[p]]--;
	if(cnt[a[p]]==0)
		cur--;
}
int main(){
	int n;
	cin>>n;
	sq=sqrt(n);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	int q;
	cin>>q;
	for(int i=1;i<=q;i++){
		cin>>Q[i].l>>Q[i].r;
		Q[i].id=i;
	}
	sort(Q+1,Q+q+1);
	int l=1,r=0;
	for(int i=1;i<=q;i++){
		while(l>Q[i].l){
			add(--l);
		}
		while(r<Q[i].r){
			add(++r);
		}
		while(l<Q[i].l){
			del(l++);
		}
		while(r>Q[i].r){
			del(r--);
		}
		ans[Q[i].id]=cur;
	}
	for(int i=1;i<=q;i++){
		cout<<ans[i]<<endl;
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值