CSUST 4000 你真的会数据结构吗?题解(暴力+质因子分解)

题目链接

题目大意

CSUST 4000 你真的会数据结构吗?题解(暴力+质因子分解)_第1张图片

题目思路

感觉这个题目不是很难啊qwq,
a[i]最多只有30,对应10个素因子,那么就可以考虑暴力qwq
然后题目的f(n)代表了 2 c n t 2^cnt 2cnt,cnt代表d的素因子个数(显然)
每次查询只要查询30次就行了,因为儿子只有最多30个不同的值

代码

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define fi first
#define se second
#define debug printf(" I am here\n");
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
const ll INF=0x3f3f3f3f3f3f3f3f;
const int maxn=1e5+5,inf=0x3f3f3f3f,mod=19940417;
const double eps=1e-10;
int n,head[maxn],a[maxn],q,cnt;
int fa[maxn];
int num[]={0,2,3,5,7,11,13,17,19,23,29};
int sta[50];
unordered_map<int,int> son[maxn];
struct node{
    int to,next;
}e[maxn<<1];
void add(int u,int v){
    e[++cnt]={v,head[u]};
    head[u]=cnt;
}
void dfs(int me,int dad){
    for(int i=head[me];i;i=e[i].next){
        if(e[i].to==dad) continue;
        son[me][sta[a[e[i].to]]]++;
        fa[e[i].to]=me;//预处理他的父亲
        dfs(e[i].to,me);
    }
}
int cal(int x){//计算x的二进制中1的个数
    int cnt=0;
    while(x){
        cnt+=(x&1);
        x=x>>1;
    }
    return cnt;
}
signed main(){
    for(int i=1;i<=30;i++){//预处理i有多少个质因子
        int sum=0;
        for(int j=1;j<=10;j++){
            if(i%num[j]==0){
                sum+=(1<<j);
            }
        }
        sta[i]=sum;
    }
    scanf("%d",&n);
    for(int i=1,u,v;i<=n-1;i++){
        scanf("%d%d",&u,&v);
        add(u,v),add(v,u);
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    dfs(1,1);
    scanf("%d",&q);
    for(int i=1,opt,u,x;i<=q;i++){
        scanf("%d%d",&opt,&u);
        if(opt==1){
            scanf("%d",&x);
            son[fa[u]][sta[a[u]]]--;
            a[u]=x;
            son[fa[u]][sta[a[u]]]++;
        }else{
            int num=0,fn=inf;
            for(int j=1;j<=30;j++){
                if(son[u][sta[j]]){
                    if(cal(sta[j])<fn){
                        fn=cal(sta[j]);
                        num=son[u][sta[j]];
                    }else if(cal(sta[j])==fn){
                        num=max(num,son[u][sta[j]]);
                    }
                }
            }
            if(fn==inf){
                printf("0 0\n");
            }else{
                 printf("%d %d\n",num,(1<<fn));
            }
        }
    }
    return 0;
}

你可能感兴趣的:(数论)