bzoj 1064: [Noi2008]假面舞会 （图论+DFS）

题目描述

传送门

题目大意：共n个人，k(k>3)种面具，只有戴第i 类面具的人才能看到戴第i+1 类面具的人的编号，戴第k 类面具的人能看到戴第1 类面具的人的编号。给出m对关系，(x,y)表示x可以看到y。现在请你根据给出的关系计算，至多和至少有多少类面具。如果k<3或者信息有误，输出-1.

题解

给出的是一个有向图，求出的k必须满足是所有环上点数的倍数，那么最直接的想法一定是找出所有的环求环上点数的gcd。
那么关键就是如何找出所有的环并计算点数(可以不考虑环套环)。
对于关系(x,y)加入x->y权值为1的有向边，然后同时加入y->x权值为-1的有向边，把图变成无向图。
然后进行dfs,如果与x相连的v已经访问过了，那么就用abs(d[x]+c[i]-d[v[i]])更新答案。
对于至少的问题，找到ans的大于3的最小因子即可。
还有就是没有环的情况，可能是多条链或者树结构。对于这种情况答案就是每个连通块的最长链的和。最小值直接就是3.

代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define M 2000003
#define N 200003
using namespace std;
int point[N],d[N],vis[N],n,m,tot,ans,mn,mx;
int nxt[M],c[M],v[M];
void add(int x,int y,int z)
{
    tot++; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
} 
int gcd(int x,int y){
    int r;
    while (y) {
        r=x%y;
        x=y; y=r;
    }
    return x;
}
void dfs(int x)
{
    vis[x]=1;
    for (int i=point[x];i;i=nxt[i])
     if (vis[v[i]]) ans=gcd(ans,abs(d[x]+c[i]-d[v[i]]));
     else {
        d[v[i]]=d[x]+c[i];
        dfs(v[i]);
     }
}
void solve(int x)
{
    vis[x]=1;
    mn=min(d[x],mn); mx=max(d[x],mx);
    for (int i=point[x];i;i=nxt[i]){
        if (vis[v[i]]) continue;
        d[v[i]]=d[x]+c[i];
        solve(v[i]);
    }
}
int main()
{
    freopen("party2008.in","r",stdin);
    freopen("party2008.out","w",stdout); 
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++) {
        int x,y; scanf("%d%d",&x,&y);
        add(x,y,1); add(y,x,-1);
    } 
    for (int i=1;i<=n;i++)
     if (!vis[i]) dfs(i);
    if (ans) {
        if (ans<3) {
            printf("-1 -1\n");
            return 0;
        }
        mn=0;
        for (int i=1;i<=ans;i++)
         if (ans%i==0&&i>=3) {
            mn=i;
            break;
         }
        printf("%d %d\n",ans,mn);
        return 0;
    }
    memset(vis,0,sizeof(vis));
    memset(d,0,sizeof(d));
    for (int i=1;i<=n;i++)
     if (!vis[i]) {
        mx=mn=0;
        solve(i);
        ans+=mx-mn+1;
     }
    mn=3;
    if (ans<3) ans=mn=-1;
    printf("%d %d\n",ans,mn);
}