hdu 6430 bitset暴力

版权声明:本文为博主原创文章,转载标明出处。 https://blog.csdn.net/xlzhang223/article/details/83868377

题意:

一棵有根树,1e5,输出每个节点的一个查询,查询定义为:以这个节点为根的树任选两个节点 权值GCD 最大 若没有孩子,则为-1

思路:

群里看见bitset居然有find_first()和find_next(x)的操作,搜了一下抄了一个大佬的思路具体实现就是 暴力子树的因子,用bitset合并,查询就在合并的时候&(保证了因子出现两次)然后find_first一下就行了。因为没有findlast,我们插入的时候反着插入就可以了 显然时间复杂度n*n/64

代码:

#include<bits/stdc++.h>
#define PB push_back
using namespace std;
const int N = 1e5+7;
vector<int> E[N];
int v[N],ans[N],n,x;

bitset<N> gao(int x){
    bitset<N> as(0),bs(0);
    for(int i=1;i*i<=v[x];i++)
        if(v[x]%i==0)as[N-i]=as[N-v[x]/i]=1;
    int ret=-1,en=E[x].size();
    for(int i=0;i<en;i++){
        bs=gao(E[x][i]);
        ret=max(ret,(int)(N-(as&bs)._Find_first()));
        as|=bs;
    }
    ans[x]=ret;
    return as;
}

int main(){
    std::ios::sync_with_stdio(false);
    cin>>n;
    for(int i=2;i<=n;i++){
        cin>>x;
        E[x].PB(i);
    }
    for(int i=1;i<=n;i++)cin>>v[i];
    gao(1);
    for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/xlzhang223/article/details/83868377