[BZOJ4602][Sdoi2016]齿轮（加权并查集）

=== ===

这里放传送门

=== ===

题解

作为SDOI R2的题这题已经很良心了。。。然而ATP当时仍然写挂了。。不是很懂自己为什么这么辣鸡(╯‵□′)╯︵┻━┻

首先可以看出这个齿轮的转动关系是具有传递性的，如果知道x和y的关系，也知道y和z的关系，就可以推出x和z的关系。具有这种关系的东西一般都可以用加权并查集来搞。对于每个元素给它维护一个dis[x]，表示的含义是如果x的代表元素转动1圈，x要转动几圈。每次读入一个限制如果x和y不在同一个集合里就进行合并。合并的时候先找到x的代表元素r1和y的代表元素r2，然后我们要把r2合并到r1上。要进行合并就要知道r1转一圈的时候r2转几圈。设从读入的信息可以知道x转一圈的时候y转w圈，那么如果r1转了一圈，y就转了$dis[x]*w$圈；而当y转了$dis[x]*w$圈的时候，r2转了$dis[x]*w \over dis[y]$圈。关系如下图：

如果读入的x和y在一个集合里，那么就可以用维护的信息算出x转了一圈的时候y转了几圈，和读入的信息进行比对就可以判定是否发生矛盾。

这个数据范围看起来好像很小的样子。。但是如果它出上n-1个信息，每次都是当齿轮i转动1圈的时候齿轮i+1转动100圈，那么最后就会发现当齿轮1转动一圈的时候齿轮n会转动$100^n$圈。。这样的话就不能直接用int或者long long或者什么类似的东西来维护加权并查集的dis了。。用分解质因数似乎是比较稳的方法，但是听说用long double还是什么类似的东西也能过？

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100
#define size 25
using namespace std;
int T,n,m,father[1010],num[110],prm[30];
bool flag;
struct info{
    int p[30],type;
    info(int X=0){
        memset(p,0,sizeof(p));
        type=1;
        if (X==0) return;
        if (X<0){type=0;X=-X;}
        else type=1;
        for (int i=1;i<=prm[0];i++){
            if (X==1) break;
            while (X%prm[i]==0){
                ++p[i];X/=prm[i];
            }
        }
    }
    info operator * (const info &a){
        info c;
        c.type=type^a.type^1;
        for (int i=1;i<=size;i++)
          c.p[i]=p[i]+a.p[i];
        return c;
    }
    info operator / (const info &a){
        info c;
        c.type=type^a.type^1;
        for (int i=1;i<=size;i++)
          c.p[i]=p[i]-a.p[i];
        return c;
    }
    bool operator == (const info &a)const{
        if (type!=a.type) return false;
        for (int i=1;i<=size;i++)
          if (p[i]!=a.p[i]) return false;
        return true;
    }
    bool operator != (const info &a)const{return !(*this==a);}
}dis[1010];
int find(int x){
    if (x==father[x]) return x;
    int tmp=find(father[x]);
    dis[x]=dis[x]*dis[father[x]];
    father[x]=tmp;
    return father[x];
}
void get_prime(){
    bool ext[110];
    memset(ext,false,sizeof(ext));
    for (int i=2;i<=100;i++){
        if (ext[i]==false) prm[++prm[0]]=i;
        for (int j=1;j<=prm[0];j++){
            if (i*prm[j]>N) break;
            ext[i*prm[j]]=true;
            if (i%prm[j]==0) break;
        }
    }
    for (int i=1;i<=prm[0];i++) num[prm[i]]=i;
}
int main()
{
    scanf("%d",&T);
    get_prime();
    for (int wer=1;wer<=T;wer++){
        scanf("%d%d",&n,&m);
        for (int i=1;i<=n;i++){
            father[i]=i;dis[i]=info();
        }
        flag=false;
        for (int i=1;i<=m;i++){
            int u,v,x,y,r1,r2;
            info tmp,X,Y;
            scanf("%d%d%d%d",&u,&v,&x,&y);
            if (flag==true) continue;
            r1=find(u);r2=find(v);
            X=info(x);Y=info(y);
            tmp=dis[u]/dis[v];
            Y=X/Y;
            if (r1!=r2){
                father[r1]=r2;
                tmp=Y/tmp;
                dis[r1]=tmp;
            }else
              if (tmp!=Y) flag=true;
        }
        printf("Case #%d: ",wer);
        if (flag==false) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}

偏偏在最后出现的补充说明

做题的时候一定要多考虑一下它的极端情况