BZOJ 2597: [Wc2007]剪刀石头布 费用流

题意

给你一个竞赛图，有的边还没被定向。要你求一种定向方案，使得简单三元环的数量最多。
n<=100

分析

果然还是我太菜。。。

注意到任选三个点，若不能组成三元环，则必然有一个点的出边数量位2，所以简单三元环的数量即为$C_n^3-\sum_{i=1}^nC_{degree[x]}^2$
考虑转化成费用流模型：
把每条边看做一个点。
s往每个点连流量为1，费用为0,1,…,n-2的边。
若一条边可能或一定作为一个点的出边，则该点往该边连流量为1费用为0的边。
每条边往t连流量为1费用为0的边。
$C_n^3-最小费用$即是答案。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int N=105;
const int inf=(int)1e9;

int n,map[N][N],cnt,last[N*N],dis[N*N],s,t,pre[N*N],num[N][N],pos[N][N];
queue<int> que;
struct edge{int to,next,c,w;}e[N*N*20];
bool vis[N*N];

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 addedge(int u,int v,int c,int w)
{
    e[++cnt].to=v;e[cnt].c=c;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].c=0;e[cnt].w=-w;e[cnt].next=last[v];last[v]=cnt;
}

bool spfa()
{
    for (int i=s;i<=t;i++) dis[i]=inf;
    dis[s]=0;vis[s]=1;que.push(s);
    while (!que.empty())
    {
        int u=que.front();que.pop();
        for (int i=last[u];i;i=e[i].next)
            if (e[i].c&&dis[u]+e[i].w<dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+e[i].w;
                pre[e[i].to]=i;
                if (!vis[e[i].to]) vis[e[i].to]=1,que.push(e[i].to);
            }
        vis[u]=0;
    }
    if (dis[t]==inf) return 0;
    else return 1;
}

int mcf()
{
    int x=t;
    while (x!=s)
    {
        e[pre[x]].c--;
        e[pre[x]^1].c++;
        x=e[pre[x]^1].to;
    }
    return dis[t];
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            map[i][j]=read();
    int sz=0;
    for (int i=1;i<n;i++)
        for (int j=i+1;j<=n;j++)
            num[i][j]=++sz;
    s=0;t=n+sz+1;cnt=1;
    for (int i=1;i<=n;i++)
        for (int j=0;j<n-1;j++)
            addedge(s,i,1,j);
    for (int i=1;i<=sz;i++) addedge(i+n,t,1,0);
    for (int i=1;i<n;i++)
        for (int j=i+1;j<=n;j++)
            if (map[i][j]==1) addedge(i,num[i][j]+n,1,0);
            else if (!map[i][j]) addedge(j,num[i][j]+n,1,0);
            else addedge(i,num[i][j]+n,1,0),addedge(j,num[i][j]+n,1,0),pos[i][j]=cnt;
    int ans=n*(n-1)*(n-2)/6;
    while (spfa()) ans-=mcf();
    printf("%d\n",ans);
    for (int i=1;i<n;i++)
        for (int j=i+1;j<=n;j++)
            if (map[i][j]==2)
            {
                if (e[pos[i][j]].c) map[i][j]=0,map[j][i]=1;
                else map[i][j]=1,map[j][i]=0;
            }
    for (int i=1;i<=n;i++)
    {
        for (int j=1;j<n;j++) printf("%d ",map[i][j]);
        printf("%d\n",map[i][n]);
    }
    return 0;
}