[BZOJ]2661: [BeiJing wc2012]连连看 费用流

Description

凡是考智商的题里面总会有这么一种消除游戏。不过现在面对的这关连连看可不是QQ游戏里那种考眼力的游戏。我们的规则是，给出一个闭区间[a,b]中的全部整数，如果其中某两个数x,y（设x>y）的平方差x2-y2是一个完全平方数z2，并且y与z互质，那么就可以将x和y连起来并且将它们一起消除，同时得到x+y点分数。那么过关的要求就是，消除的数对尽可能多的前提下，得到足够的分数。快动手动笔算一算吧。

题解：

直接上费用流，一开始太菜忽略了直接建单向边的话点会被重复选择，建双向边的话就可以保证每个点只被选择一次了。

代码：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;
const int inf=999999999;
const int maxn=4010;
int st,ed,n,m;
struct edge{int x,y,d,c,next;}e[2110000];
int last[maxn],len=1;
void ins(int x,int y,int d,int c)
{
    int t=++len;
    e[t].x=x;e[t].y=y;e[t].next=last[x];last[x]=t;
    e[t].d=d;e[t].c=c;
    t=++len;
    e[t].x=y;e[t].y=x;e[t].next=last[y];last[y]=t;
    e[t].d=0;e[t].c=-c;
}
int pre[maxn],f[maxn];
bool in[maxn];
int ans=0;
queue<int>q;
bool spfa()
{
    memset(f,63,sizeof(f));f[st]=0;
    memset(in,false,sizeof(in));in[st]=true;
    q.push(st);pre[st]=-1;
    while(!q.empty())
    {
        int x=q.front();q.pop();in[x]=false;
        for(int i=last[x];i;i=e[i].next)
        {
            int y=e[i].y;
            if(e[i].d&&f[x]+e[i].c<f[y])
            {
                pre[y]=i;
                f[y]=f[x]+e[i].c;
                if(!in[y])q.push(y),in[y]=true;
            }
        }
    }
    if(f[ed]==1061109567)return false;
    return true;
}
void work()
{
    int mind=inf,x=pre[ed];
    while(x!=-1)
    {
        mind=min(mind,e[x].d);
        x=pre[e[x].x];
    }
    x=pre[ed];
    while(x!=-1)
    {
        e[x].d-=mind;e[x^1].d+=mind;
        ans+=e[x].c*mind;
        x=pre[e[x].x];
    }
}
int gcd(int a,int b)
{
    if(b==0)return a;
    return gcd(b,a%b);
}
bool mark[1000010];
int main()
{
    for(int i=1;i<=1000;i++)mark[i*i]=true;
    scanf("%d%d",&n,&m);
    st=2*m+1;ed=2*m+2;
    for(int i=n+1;i<=m;i++)
    for(int j=n;j<i;j++)
    if(mark[i*i-j*j]&&gcd(j,(int)(sqrt(i*i-j*j)))==1)
    {
        ins(i,j+m,1,-i-j);
        ins(j,i+m,1,-i-j);
    }
    for(int i=n;i<=m;i++)ins(st,i,1,0),ins(i+m,ed,1,0);
    int ans1=0;
    while(spfa())work(),ans1++;
    printf("%d %d",ans1/2,-ans/2);
}