2661: [BeiJing wc2012]连连看

题目链接

题目大意：给出一个闭区间[a,b]中的全部整数，如果其中某两个数x,y（设x>y）的平方差x2-y2是一个完全平方数z2，并且y与z互质，那么就可以将x和y连起来并且将它们一起消除，同时得到x+y点分数。求最大得分

题解：Orz 题解

我的收获：拆点的原则？（

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

const int M=2050;
const int INF=1e9;

int n,L,R,id[M];
int flow,mincost;
int t,head[M],last[M];
int st,ed;
int d[M],pre[M];
bool vis[M];

queue<int> q;

struct edge{int fro,to,nex,c,val;}e[M*M];

void add(int i,int j,int ca,int co){e[t].fro=i,e[t].to=j,e[t].nex=head[i],e[t].c=ca,e[t].val=co;head[i]=t++;}
void insert(int i,int j,int w,int z){add(i,j,w,z),add(j,i,0,-z);}

bool spfa()
{
    for(int i=0;i<=ed;i++) vis[i]=0,d[i]=INF;
    d[st]=0;q.push(st);
    while(!q.empty())
    {
        int u=q.front();q.pop();vis[u]=false;
        for(int i=head[u];i!=-1;i=e[i].nex){
            int v=e[i].to;
            if(e[i].c&&d[v]>d[u]+e[i].val){
                d[v]=d[u]+e[i].val;
                pre[v]=i;
                if(!vis[v]){
                    vis[v]=true;
                    q.push(v);
                }
            }
        }
    }
    return d[ed]!=INF;
}

void mcfx()
{
    int mx=INF;
    for(int u=ed;u!=st;u=e[pre[u]].fro)
        mx=min(mx,e[pre[u]].c);
    for(int u=ed;u!=st;u=e[pre[u]].fro){
        e[pre[u]].c-=mx;e[pre[u]^1].c+=mx;
        mincost+=mx*e[pre[u]].val;
    }
    flow+=mx;
}

void work()
{
    while(spfa()) mcfx();
    printf("%d %d\n",flow/2,-mincost/2);
}

void build()
{
    for(int i=1;i<=n;i++) insert(st,i,1,0),insert(i+n,ed,1,0);
    for(int i=L;i<=R;i++){
        for(int j=L;j<i;j++)
        {
             int sq=sqrt(i*i-j*j);
             if(sq*sq==i*i-j*j&&__gcd(j,sq)==1) insert(id[i],id[j]+n,1,-i-j),insert(id[j],id[i]+n,1,-i-j);
        }
    }
}

void init()
{
    scanf("%d%d",&L,&R);
    for(int i=L;i<=R;i++) id[i]=++n;
    memset(head,-1,sizeof(head));
    st=0,ed=n+n+1;
    build();
}

int main()
{
    init();
    work();
    return 0;
}