codeforces 842C dp

题目大意:给你一棵树,树上的每一个节点有一个值,定义一个节点的美丽值为这个节点到根节点的的路径上的每一个个节点(包括自己)的所有的值的最大公约数,现在给定一种操作,对于某一个点的美丽值现在可以把路径上的任意一个点的值变为0,(规定GCD(x,0)=x)问对于每一个个点单独考虑,输出每一个点的美丽值的最大值。


思路:在大佬的帮助下明白了dp的思路,定义dp[i][0]表示从根节点到i已经有一个被改变为0的所有可能产生的公约数的集合,dp[i][1]表示从根节点到i没有节点被变为0的所有可能的公约数的集合,这里公约数的集合只可能是logn的级别,具体的原因我还不是很清楚。
那么很明显dp应该这样定义 set<int>dp[N][2]
转移很简单,这里就不写了,直接看代码

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<cstring>
#define clr(x) memset(x,0,sizeof(x))
using namespace std;
#define LL long long
int dat[200005];
vector<int>vec[200005];
int ans[200005];
bool vis[200005];
set<int>dp[200005][2];
int gcd(int a,int b)
{
    if(a==0||b==0)return max(a,b);
    return a%b==0?b:gcd(b,a%b);
}
void dfs(int now,int ma)
{
    for(set<int>::iterator it = dp[now][0].begin();it!=dp[now][0].end();it++)
    {
        ans[now] = max(ans[now],*it);
    }
    for(set<int>::iterator it = dp[now][1].begin();it!=dp[now][1].end();it++)
    {
        ans[now] = max(ans[now],*it);
    }
    vis[now] = true;
    for(int i = 0;i<vec[now].size();i++)
    {
        if(vis[vec[now][i]])continue;

        for(set<int>::iterator it = dp[now][1].begin();it!=dp[now][1].end();it++)
        {

            int temp = gcd(*it,dat[vec[now][i]]);
            //cout << "k" << temp <<" " << *it <<vec[]<< endl;
            dp[vec[now][i]][0].insert(*it);
            dp[vec[now][i]][1].insert(temp);
        }
        for(set<int>::iterator it = dp[now][0].begin();it!=dp[now][0].end();it++)
        {
            int temp = gcd(*it,dat[vec[now][i]]);
            dp[vec[now][i]][0].insert(temp);
            //dp[vec[now][i]][1].insert(temp);
        }
        dfs(vec[now][i],ans[now]);
    }
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        clr(vis);
        clr(ans);
        for(int i = 0;i<=n;i++)
            vec[i].clear(),dp[i][0].clear(),dp[i][1].clear();
        for(int i = 1;i<=n;i++)
            scanf("%d",&dat[i]);
        for(int i = 0;i<n-1;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            vec[a].push_back(b);
            vec[b].push_back(a);
        }
        dp[1][1].insert(dat[1]);
        dp[1][0].insert(0);
        dfs(1,0);
        //for(set<int>::iterator it = dp[2][0].begin();it!=dp[2][0].end();it++)cout << *it <<" ";cout << endl;
        for(int i = 1;i<=n;i++)
        {
            if(i!=1)printf(" ");
            printf("%d",ans[i]);
        }printf("\n");
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值