bzoj 2661(最大费用流)

2661: [BeiJing wc2012]连连看

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 973   Solved: 390
[ Submit][ Status][ Discuss]

Description

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

Input

        
 只有一行，两个整数，分别表示a,b。

Output

 两个数，可以消去的对数，及在此基础上能得到的最大分数。

Sample Input

1 15

Sample Output

2 34

HINT

对于30%的数据，1<=a,b<=100

对于100%的数据，1<=a,b<=1000

解题思路：普通的费用流，不过是最大费用

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>
using namespace std;
int a,c,len,S,T,n,ans1,ans2;
int from[5000000],to[5000000],wh[5000000],next[5000000],f[5000000];
int h[3010],dis[3010],pre[3010],q[1000000];
bool b[3010];


inline int read()
{
char y; int x=0,f=1; y=getchar();
while (y<'0' || y>'9') {if (y=='-') f=-1; y=getchar();}
while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
return x*f;
}


int gcd(int x,int y)
 {
  if (y==0) return x;else return gcd(y,x%y);
 }


void ins(int a,int b,int flow,int wg)
 {
  ++len; from[len]=a; to[len]=b; f[len]=flow; wh[len]=wg; next[len]=h[a]; h[a]=len;
 }


void insert(int a,int b,int flow,int wg)
 {
  ins(a,b,flow,wg); ins(b,a,0,-wg);
 }


bool jud(int a,int b)
 {
  bool mg;
  if (a<b)swap(a,b);
  double sum=int(sqrt(a*a-b*b))*int(sqrt(a*a-b*b));
  if (sum==a*a-b*b) mg=true;else mg=false;
  if (gcd(a,b)==1 && mg) return true;else return false;
 }


bool spfa()
 {
  memset(dis,-0x7f,sizeof(dis)); memset(b,true,sizeof(b));
  dis[S]=0; b[S]=false; int tail=0,head=0; q[++tail]=S;
  while (head<tail)
  {
  ++head;
  int u=h[q[head]];
  while (u!=0)
  {
    if (f[u]>0 && dis[to[u]]<dis[q[head]]+wh[u])
{
     pre[to[u]]=u;
 dis[to[u]]=dis[q[head]]+wh[u];
 if (b[to[u]])
  {
  b[to[u]]=false; ++tail; q[tail]=to[u];
  }
} 
    u=next[u]; 
}
 }
if (dis[T]>0)return true;else return false;
 } 


void getans()
 {
  int now=T; int mx=0x7fffffff;
  while (now!=S)
  {
  mx=min(mx,f[pre[now]]);
  now=from[pre[now]];
}
now=T; ans1+=mx;
while (now!=S)
{
ans2+=mx*wh[pre[now]]; 
f[pre[now]]-=mx; f[pre[now]^1]+=mx;
now=from[pre[now]];
}
 }


int main()
{
S=3001; T=3002;
  a=read(); c=read(); n=c-a+1; int st=a-1; len=1;
  for (int i=a;i<=c;++i)
  for (int j=a;j<=c;++j)
   if (i!=j)
    {
     if (jud(i,j))
 insert(i-st,j-st+n,1,i+j),insert(j-st,i-st+n,1,i+j);
  }
   for (int i=1;i<=n;++i)
    {
      insert(S,i,1,0); insert(i+2*n,T,1,0);
      insert(i+n,i+2*n,1,0);
}
   while (spfa())
    {
    getans();
}
   printf("%d %d",ans1/2,ans2/2);
}