codeforces 463E Caisa and Tree 栈+dfs

题意:对于一棵树,1为根结点,每个结点有一个值,有两种操作:1.将从根结点到结点v的路径中找最接近v的某个结点,满足两个点

对应值的gcd不为1 。  2.将结点v的值修改为w。


思路:我们从根开始dfs,对于当前访问的u,我们其对应的值a[u]分解成质因数的形式,将它在st中的位置存入map<int, stack<int>>

mp中。其中,键值为a [ u ]的质因数。不妨设x为a[ u ]的某个质因数,如果mp[x]的栈中已经有值,说明是从根结点到u的路径中有

和u有最大公约数不为1的值存在,而且那个值在st中的位置就是栈元素所代表的。 所以对于a[ u ]的所有质因数对应的栈中元素取最

大值就是最靠近u的与a[ u ]的gcd不为1的值位置。当u及其子树全部访问完之后我们把st、mp[x](x为a[ u ]的质因数)与u有关的全部

清除这样经过一次dfs之后,对于cmd=1来说,就可以直接O(1)查询。 对于cmd=2,由于改变的树某些结点的值,所以需要重新

dfs详见代码:

// file name: codeforces463E.cpp //
// author: kereo //
// create time:  2014年08月30日 星期六 22时02分29秒 //
//***********************************//
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
const int MAXN=100000+100;
const int inf=0x3fffffff;
#define L(x) (x<<1)
#define R(x) (x<<1|1)
int n,Q,edge_cnt;
int a[MAXN],head[MAXN],ans[MAXN];
map<int,stack<int> >mp;
vector<int>fac[MAXN];
vector<int>st;
struct Edge
{
	int v;
	int next;
}edge[MAXN*2];
void init(){
	edge_cnt=0;
	memset(head,-1,sizeof(head));
}
void addedge(int u,int v){
	edge[edge_cnt].v=v;
	edge[edge_cnt].next=head[u];
	head[u]=edge_cnt++;
}
vector<int> factorize(int x){  //将x分解成质因数
	vector<int>vec;
	for(int i=2;i*i<=x;i++){
		if(x % i == 0)
			vec.push_back(i);
		while(x % i == 0)
			x/=i;
	}
	if(x>1) vec.push_back(x);
	return vec;
}
int top(int x){
	if(!mp.count(x) || mp[x].empty()) return -1;
	return mp[x].top();
}
void dfs(int u,int fa){
	int Max=-1;
	for(int i=0;i<fac[u].size();i++)
		Max=max(Max,top(fac[u][i])); //选最靠后的
	ans[u]=Max == -1 ? -1 : st[Max]; //最靠后位置的值
	st.push_back(u);
	for(int i=0;i<fac[u].size();i++)
		mp[fac[u][i]].push(st.size()-1);
	for(int i=head[u];i!=-1;i=edge[i].next){
		int v=edge[i].v; 
		if(v == fa) continue;
		dfs(v,u);
	}
	st.pop_back(); //访问完u,要clear
	for(int i=0;i<fac[u].size();i++)
		mp[fac[u][i]].pop();
}
int main()
{
	while(~scanf("%d%d",&n,&Q)){
		init();
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			fac[i]=factorize(a[i]);
		}
		for(int i=1;i<n;i++){
			int u,v;
			scanf("%d%d",&u,&v);
			addedge(u,v); addedge(v,u);
		}
		dfs(1,1);
		while(Q--){
			int cmd;
			scanf("%d",&cmd);
			if(cmd == 1){
				int x;
				scanf("%d",&x);
				printf("%d\n",ans[x] == -1 ? -1 : ans[x]);
			}
			if(cmd == 2){
				int x,w;
				scanf("%d%d",&x,&w);
				a[x]=w;
				fac[x]=factorize(w);
				dfs(1,1);
			}
		}
	}
	return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值