HDU - 4902 Nice boat(线段树)

题目链接:点击查看

题目大意:给出 n 个数字组成的数列 a ,需要进行 m 次操作,每次操作分为以下两种:

  1. 1 l r val:将区间 [ l , r ] 内的 a[ i ] 都赋值为 val
  2. 2 l r val:将区间 [ l , r ] 内大于 val 的 a[ i ] 都赋值为 gcd( a[ i ] , val )

最后输出整个序列

题目分析:因为题目很良心的给了 15 秒,所以可以直接上手写,写一点暴力剪枝就好了,具体剪枝方法如下:

每个节点维护一个 mmax 最大值,在执行操作 2 的时候,如果碰到 mmax <= val 时,就可以直接返回了

还有一个剪枝方法是,维护一个 same 变量,如果当前节点的所有数字都相等的话,那么 same 为数字的值,否则为 -1 ,在执行操作二时,如果遍历到 same != -1 的结点时,直接区间更新此结点然后返回就好了

对于操作一没什么可说的,维护一个 lazy 标记中规中矩的写就好了

代码:
 

#include<iostream>
#include<cstdio>
#include<string>
#include<ctime>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<stack>
#include<climits>
#include<queue>
#include<map>
#include<set>
#include<sstream>
#include<cassert>
using namespace std;

typedef long long LL;

typedef unsigned long long ull;

const int inf=0x3f3f3f3f;

const int N=1e5+100;

struct Node
{
	int l,r,lazy,mmax,same;
}tree[N<<2];

void pushup(int k)
{
	tree[k].mmax=max(tree[k<<1].mmax,tree[k<<1|1].mmax);
	if(tree[k<<1].same==tree[k<<1|1].same)
		tree[k].same=tree[k<<1].same;
	else
		tree[k].same=-1;
}

void pushdown(int k)
{
	if(tree[k].lazy!=-1)
	{
		int lz=tree[k].lazy;
		tree[k<<1].mmax=tree[k<<1].lazy=tree[k<<1].same=lz;
		tree[k<<1|1].mmax=tree[k<<1|1].lazy=tree[k<<1|1].same=lz;
		tree[k].lazy=-1;
	}
}

void build(int k,int l,int r)
{
	tree[k].l=l;
	tree[k].r=r;
	tree[k].lazy=-1;
	if(l==r)
	{
		scanf("%d",&tree[k].mmax);
		tree[k].same=tree[k].mmax;
		return;
	}
	int mid=l+r>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	pushup(k);
}

void change1(int k,int l,int r,int val)//将[l,r]变为val 
{
	if(tree[k].r<l||tree[k].l>r)
		return;
	if(tree[k].l>=l&&tree[k].r<=r)
	{
		tree[k].mmax=tree[k].lazy=tree[k].same=val;
		return;
	}
	pushdown(k);
	change1(k<<1,l,r,val);
	change1(k<<1|1,l,r,val);
	pushup(k);
}

void change2(int k,int l,int r,int val)//将[l,r]大于val的变为 gcd(val,a[i]) 
{
	if(tree[k].mmax<=val)//剪枝 
		return;
	if(tree[k].r<l||tree[k].l>r)
		return;
	if(tree[k].l>=l&&tree[k].r<=r)
	{
		if(tree[k].same!=-1)
		{
			tree[k].mmax=tree[k].lazy=tree[k].same=__gcd(tree[k].mmax,val);
			return;
		}
	}
	pushdown(k);
	change2(k<<1,l,r,val);
	change2(k<<1|1,l,r,val);
	pushup(k);
}

int query(int k,int pos)
{
	if(tree[k].l==tree[k].r)
		return tree[k].mmax;
	pushdown(k);
	int mid=tree[k].l+tree[k].r>>1;
	if(pos<=mid)
		return query(k<<1,pos);
	else
		return query(k<<1|1,pos);
}
 
int main()
{
#ifndef ONLINE_JUDGE
//  freopen("input.txt","r",stdin);
//  freopen("output.txt","w",stdout);
#endif
//  ios::sync_with_stdio(false);
	int w;
	cin>>w;
	while(w--)
	{
		int n;
		scanf("%d",&n);
		build(1,1,n);
		int m;
		scanf("%d",&m);
		while(m--)
		{
			int op,l,r,val;
			scanf("%d%d%d%d",&op,&l,&r,&val);
			if(op==1)
				change1(1,l,r,val);
			else
				change2(1,l,r,val);
		}
		for(int i=1;i<=n;i++)
			printf("%d ",query(1,i));
		puts("");
	}



























    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_45458915/article/details/106947158