关闭

[CodeM初赛A轮]D

290人阅读 评论(3) 收藏 举报
分类:

题解

用sqrt(a)/log a的时间分解质因数。
枚举一个质因数x,所有x的倍数节点打标记。
对于每一个被标记的联通块求直径。

#include<cstdio>
#include<algorithm>
#include<map>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=100000+10,maxd=32000+10;
map<int,int> id;
int h[maxn],go[maxn*2],nxt[maxn*2],a[maxn],b[maxn*11],f[maxn];
int h2[maxn*11],g2[maxn*11],n2[maxn*11];
int pri[maxd];
bool bz[maxd],pd[maxn],vis[maxn];
int i,j,k,l,t,n,m,tot,top,num,cnt,ans,mx,now;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
void prepare(){
    fo(i,2,maxd-10){
        if (!bz[i]) pri[++top]=i;
        fo(j,1,top){
            if ((ll)i*pri[j]>maxd-10) break;
            bz[i*pri[j]]=1;
            if (i%pri[j]==0) break;
        }
    }
}
void add(int x,int y){
    go[++tot]=y;
    nxt[tot]=h[x];
    h[x]=tot;
}
void add2(int x,int y){
    g2[++cnt]=y;
    n2[cnt]=h2[x];
    h2[x]=cnt;
}
void dfs(int x,int y,int z){
    vis[x]=1;
    if (z>now) mx=x,now=z;
    int t=h[x];
    while (t){
        if (go[t]!=y&&pd[go[t]]) dfs(go[t],x,z+1);
        t=nxt[t];
    }
}
int main(){
    prepare();
    n=read();
    fo(i,1,n-1){
        j=read();k=read();
        add(j,k);add(k,j);
    }
    tot=0;
    fo(i,1,n){
        a[i]=read();
        k=a[i];
        fo(j,1,top){
            if ((ll)pri[j]*pri[j]>k) break;
            if (k%pri[j]==0){
                if (id[pri[j]]==0) id[pri[j]]=++tot,b[++num]=pri[j];
                add2(id[pri[j]],i);
                while (k%pri[j]==0) k/=pri[j];
            }
        }
        if (k>1){
            if (id[k]==0) id[k]=++tot,b[++num]=k;
            add2(id[k],i);
        }
    }
    fo(i,1,num){
        t=h2[id[b[i]]];
        while (t){
            pd[g2[t]]=1;
            t=n2[t];
        }
        t=h2[id[b[i]]];
        while (t){
            if (!vis[g2[t]]){
                mx=0;now=-1;
                dfs(g2[t],0,0);
                now=-1;
                dfs(mx,0,0);
                ans=max(ans,now);
            }
            t=n2[t];
        }
        t=h2[id[b[i]]];
        while (t){
            vis[g2[t]]=0;
            t=n2[t];
        }
        t=h2[id[b[i]]];
        while (t){
            pd[g2[t]]=0;
            t=n2[t];
        }
    }
    printf("%d\n",ans+1);
}
0
0
查看评论
发表评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场

美团CodeM 初赛 A伦 最长树链 xjb枚举+dfs

[编程题] 最长树链 时间限制:1秒 空间限制:32768K 树链是指树里的一条路径。美团外卖的形象代言人袋鼠先生最近在研究一个特殊的最长树链问题。现在树中的每个点都有一...
  • qq_30927651
  • qq_30927651
  • 2017-06-19 16:48
  • 377

[LOJ#6159][美团 CodeM 初赛 Round A][暴力即正解]最长树链

枚举每一个质数,权值是这个质数的倍数的节点设为1,否则为0,那么最长长链的长度就是点权为1的点构成的最长的链。当然不用枚举所以质数,因为每个数最多有log个不同的质因数,所以只要把所有点权质因数分解,...
  • Coldef
  • Coldef
  • 2017-07-06 00:47
  • 340

CodeM资格赛 Round A 最长树链

按照题解的做法,对于每一个质约数分别进行讨论最长链就行 对于每一个数的质约数可是比logn还要小的比赛的时候没人写,我也没看 = =,可惜了,不过我当时对于复杂度的把握也不大啊#include #...
  • basasuya
  • basasuya
  • 2017-07-01 23:47
  • 81

[两道递推题] 美团 CodeM 初赛 Round A 二分图染色 & OEIS A001499

好久没做过n=107n=10^7这种正常的递推题了二分图染色转化为棋盘模型,即 N×NN\times N 棋盘上放黑白棋子,每个格子至多放一个,同行同列没有相同颜色的棋子。 令bnb_n为只有一种颜...
  • u014609452
  • u014609452
  • 2017-06-26 20:44
  • 339

「美团 CodeM 初赛 Round A」二分图染色 容斥原理

题目链接点这里 首先绿色不用管 然后只有2种颜色,很显然需要用容斥原理求,先算全部的数量,然后容斥减去不合法的数目 我们n个点设单种颜色的方案数为F(n),所以2种颜色的方案数就是F(n)^2(...
  • qq_30927651
  • qq_30927651
  • 2017-06-21 13:22
  • 279

[CodeM初赛A轮]B

题解设dp f[i1,i2,j1,j2]表示选出A串的[i1,i2]以及B串的[j1,j2]能否组成回文串。 dp转移显然。 但是边界比较麻烦。 考虑先做dp fa[i,j]和fb[i,j]分表...
  • WerKeyTom_FTD
  • WerKeyTom_FTD
  • 2017-06-18 22:15
  • 424

[CodeM初赛A轮]E

题解大小点分治。 出现次数大于阈值的,处理前缀和,然后暴力枚举一个询问判断。 出现次数小于阈值的,跑莫队,维护每个数出现次数,以及每种出现次数的数的个数,然后询问可以暴力枚举出现次数。#inclu...
  • WerKeyTom_FTD
  • WerKeyTom_FTD
  • 2017-06-18 22:20
  • 383

「美团 CodeM 初赛 Round A」身体训练

题目链接:身体训练题目大意:n个人排成一列跑步,前后两人之间相隔 u 米,每个人正常速度均为 v 米/秒。 当某个配送员排在最后的时候,他需要以当时自己的最高速度往前跑,直到超过排头的人 u 米,然后...
  • Nemaleswang
  • Nemaleswang
  • 2017-07-03 20:50
  • 182

codeM初赛B轮A题

[编程题] 黑白树 时间限制:1秒 空间限制:32768K 一棵n个点的有根树,1号点为根,相邻的两个节点之间的距离为1。树上每个节点i对应一个值k[i]。每个点都有一个颜色,初始的...
  • u010770184
  • u010770184
  • 2017-06-27 11:25
  • 104

[LOJ#6159][美团 CodeM 初赛 Round A][暴力即正解]最长树链

枚举每一个质数,权值是这个质数的倍数的节点设为1,否则为0,那么最长长链的长度就是点权为1的点构成的最长的链。当然不用枚举所以质数,因为每个数最多有log个不同的质因数,所以只要把所有点权质因数分解,...
  • Coldef
  • Coldef
  • 2017-07-06 00:47
  • 340
    个人资料
    • 访问:327186次
    • 积分:11634
    • 等级:
    • 排名:第1512名
    • 原创:814篇
    • 转载:4篇
    • 译文:0篇
    • 评论:201条
    最新评论
    文章分类