cogs 魔术球问题(简化版)

1. [网络流24题]魔术球问题（简化版
   ★★☆ 输入文件：balla.in 输出文件：balla.out 简单对比
   时间限制：1 s 内存限制：128 MB
   问题描述：
   假设有n根柱子，现要按下述规则在这n根柱子中依次放入编号为 1，2，3，4……的球。
   （1）每次只能在某根柱子的最上面放球。
   （2）在同一根柱子中，任何2个相邻球的编号之和为完全平方数。
   试设计一个算法，计算出在n根柱子上最多能放多少个球。例如，在4 根柱子上最多可
   放11个球。
   ´编程任务：
   对于给定的n，计算在 n根柱子上最多能放多少个球。
   ´数据输入：
   文件第1 行有 1个正整数n，表示柱子数。
   ´结果输出:
   文件的第一行是球数。
   数据规模
   n<=60 保证答案小于1600
   输入文件示例
   4
   输出文件示例
   11
   方案如下
   1 8
   2 7 9
   3 6 10
   4 5 11
   每一行表示一个柱子上的球

---

【分析】
重要定理:最小路径覆盖=节点数-最大匹配
…于是拆点为二分图…blabla…
这破题想了老长时间orz

---

【代码】

//cogs 魔术球问题(简化版) 
#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<cmath>
#define inf 1e9+7
#define p(i,j) (i-1)*n+j
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
queue <int> q;
const int mxn=20005;
int n,m,cnt,ans,s,t,tot;
int head[mxn],dis[mxn],a[mxn];
int dx[8]={-2,-2,-1,-1,1,1,2,2},dy[8]={-1,1,-2,2,2,-2,1,-1};
struct node {int to,next,flow;} f[mxn<<2];
inline void add(int u,int v,int flow)
{
    f[++cnt].to=v,f[cnt].next=head[u],f[cnt].flow=flow,head[u]=cnt;
    f[++cnt].to=u,f[cnt].next=head[v],f[cnt].flow=0,head[v]=cnt;
}
inline bool bfs()
{
    int i,j,u,v,flow;
    memset(dis,-1,sizeof dis);
    q.push(s);
    dis[s]=0;
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        for(i=head[u];i;i=f[i].next)
        {
            v=f[i].to,flow=f[i].flow;
            if(dis[v]==-1 && flow>0)
              dis[v]=dis[u]+1,q.push(v);
        }
    }
    if(dis[t]>0) return 1;
    return 0;
}
inline int find(int u,int low)
{
    int v,i,j,a=0,sum=0,flow;
    if(u==t) return low;
    for(i=head[u];i;i=f[i].next)
    {
        v=f[i].to,flow=f[i].flow;
        if(flow>0&& dis[v]==dis[u]+1 && (a=find(v,min(low-sum,flow))))
        {
            f[i].flow-=a;
            if(i&1) f[i+1].flow+=a;
            else f[i-1].flow+=a;
            sum+=a;
        }
    }
    if(!sum) dis[u]=-1;
    return sum;
}
int main()
{
    freopen("balla.in","r",stdin);
    freopen("balla.out","w",stdout);
    int i,j,k;
    scanf("%d",&n);
    fo(i,1,60) a[i]=i*i;
    s=0,t=10000;
    fo(i,1,2000)
    {
        add(s,i,1),add(i+2000,t,1);
        fo(k,1,60)
          if(a[k]-i>0 && a[k]-i<i)
            add(a[k]-i,i+2000,1);
        while(bfs()) ans+=find(s,inf);
        if(i-ans>n)
        {
            printf("%d\n",i-1);
            return 0;
        }
    }
    return 0;
}