网络流24题04 魔术球问题

Description
假设有n根柱子，现要按下述规则在这n根柱子中依次放入编号为1，2，3，4的球。
（1）每次只能在某根柱子的最上面放球。
（2）在同一根柱子中，任何2个相邻球的编号之和为完全平方数。
试设计一个算法，计算出在n根柱子上最多能放多少个球。例如，在4 根柱子上最多可放11 个球。

编程任务：
对于给定的n，计算在n根柱子上最多能放多少个球。

Input Format
文件第1 行有1个正整数n，表示柱子数。

Output Format
程序运行结束时，将n 根柱子上最多能放的球数以及相应的放置方案输出。文件的第一行是球数。接下来的n行，每行是一根柱子上的球的编号。

Sample Input
C++

4
1
4
Sample Output
C++

11
1 8
2 7 9
3 6 10
4 5 11
1
2
3
4
5
11
1 8
2 7 9
3 6 10
4 5 11

分析：一开始很容易想到直接对于两个符合条件的编号连流量为1的边，建超级源和超级汇。然后枚举答案，看是否符合条件，跑dinic就可以了，注意是每次用ans-=dfs(0,inf)，ans表示用的柱子，明显当你用的柱子数量>n时，就不能跑了。方案的话记录一下反向边就好。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e5+5;
const int inf=2147483646;
int n,m;
typedef long long ll;
int dis[10001],cur[N],head[N],go[N],next[N],tot=1,T=10000;
bool mark[N];
int val[N],to[N];
int q[N],ans,s;
inline void add(int x,int y,int z)
{
    go[++tot]=y;
    next[tot]=head[x];
    val[tot]=z;
    head[x]=tot;
}
inline void ins(int x,int y,int z)
{
    add(x,y,z);
    add(y,x,0);
}
inline bool bfs()
{
    int t=0,w=1;
    memset(dis,-1,sizeof(dis));
    dis[0]=0;
    q[1]=0;
    while (t<w)
    {
        int x=q[++t];
        for(int i=head[x];i;i=next[i])
        {
            int v=go[i];
            if(val[i]&&dis[v]==-1)
            {
                dis[v]=dis[x]+1;
                q[++w]=v;
            } 
        }
    } 
    return dis[T]!=-1;
}
inline int dfs(int x,int f)
{
    if (x==T)return f;
    int w,used=0;
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (val[i]&&dis[v]==dis[x]+1)
        {
            w=f-used;
            w=dfs(v,min(w,val[i]));
            val[i]-=w;
            val[i^1]+=w;
            used+=w;
            if (used==f)return f;           
        }
    }
    if (!used)dis[x]=-1;
    return used;
}
inline void dinic()
{
    while (bfs())
    {
        ans-=dfs(0,inf);
    }
}
int main()
{
    scanf("%d",&n);
    while (1)
    {
        ans++,s++;
        fo(i,1,s-1)
        if (sqrt(i+s)==(int)(sqrt(i+s)))
        ins(i,s+5000,1);
        ins(0,s,1);
        ins(s+5000,T,1);
        dinic();
        if (ans>n)break;
    }
    printf("%d\n",s-1);
    //return 0;
    fo(i,1,s-1)
    {

        for(int j=head[i];j;j=next[j])
        {
            if (!val[j])
            {
                to[i]=go[j]-5000;
                break;
            }
        }
    }
    fo(i,1,s-1)
    {
        if (mark[i])continue;
        int t=i;
        while (t!=-5000)
        {
            mark[t]=1;
            printf("%d ",t);
            t=to[t];
        }
        printf("\n");
    }
    return 0;
}