网络流24题4 魔术球问题

Description

---

假设有n根柱子，现要按下述规则在这n根柱子中依次放入编号为1，2，3，…的球。

1. 每次只能在某根柱子的最上面放球。

2. 在同一根柱子中，任何2个相邻球的编号之和为完全平方数。

试设计一个算法，计算出在n根柱子上最多能放多少个球。例如，在4 根柱子上最多可放11 个球。

Analysis

---

经过再三地读题之后可以发现

1. 球的数量是未知的
2. 随着球的数量增加，所需要的柱子是非递减的

这些就够了，那么我们可以考虑枚举球的个数然后对编号和为平方数的球连边，求出的最小路径覆盖就是柱子数量所以找到第一个路径覆盖大于n的-1就是答案
求方案数我们建一个新图然后重新一遍最大流匹配求方案就行了

Code

---

/*
ID:wjp13241
PROG:
LANG:C++
*/
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <ctime>
#include <algorithm>
#include <iostream>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <map>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define dfo(i,a,b) for(int i=a;i>=b;i--)
#define fore(i,x,e) for(int i=ls[x];i;i=e[i].next)
#define fil(x,t) memset(x,t,sizeof(x))
#define FILEIN(s) freopen(s,"r",stdin)
#define FILEOUT(s) freopen(s,"w",stdout)
#define STP system("pause")
#define min(x,y) x<y?x:y
#define max(x,y) x>y?x:y
#define MP(x,y) make_pair(x,y)
#define PuB(v,x) v.push_back(x)
#define PoB(v) v.pop_back()
#define ld long double
#define ll long long
#define db double
#define INF 0x3f3f3f3f
#define LIM 5001
#define EPS 1e-4
#define N 100201
#define H 40000
#define E N*20+1
#define L 100
using namespace std;
struct edge{int y,w,rev,next;}e[E];
int dis[N],lnk[N],ls[N],v[N],maxE=0;
int add(int x,int y,int w){
    e[++maxE]=(edge){y,w,maxE+1,ls[x]};ls[x]=maxE;
    e[++maxE]=(edge){x,0,maxE-1,ls[y]};ls[y]=maxE;
}
int bfs(int st,int ed){
    fil(dis,63);
    queue<int>q;
    q.push(st);
    dis[st]=0;
    while (!q.empty()){
        int now=q.front();q.pop();
        for (int i=ls[now];i;i=e[i].next){
            if (e[i].w>0&&dis[now]+1<dis[e[i].y]){
                dis[e[i].y]=dis[now]+1;
                q.push(e[i].y);
                if (e[i].y==ed)
                    return 1;
            }
        }
    }
    return 0;
}
int find(int now,int ed,int mn){
    if (now==ed)
        return mn;
    for (int i=ls[now];i;i=e[i].next){
        if (e[i].w>0&&dis[now]+1==dis[e[i].y]){
            int d=find(e[i].y,ed,min(mn,e[i].w));
            if (d>0){
                e[i].w-=d;
                e[e[i].rev].w+=d;
                lnk[now]=e[i].y;
                return d;
            }
        }
    }
    return 0;
}
int main(){
    ios::sync_with_stdio(false);
    fo(i,1,L)
        v[i*i]=1;
    int l;
    cin>>l;
    int mxFlow=0,ans;
    fo(n,1,INF){
        add(0,n,1);
        add(n+LIM,H,1);
        fo(i,1,n-1)
            if (v[i+n])
                add(i,n+LIM,1);
        if (bfs(0,H))
            mxFlow+=find(0,H,INF);
        if (n-mxFlow>l){
            ans=n-1;
            break;
        }
    }
    cout<<ans<<endl;
    fil(lnk,0);
    fil(ls,0);
    maxE=mxFlow=0;
    fo(i,1,ans){
        add(0,i,1);
        add(i+LIM,H,1);
    }
    fo(j,1,ans)
        fo(i,1,j-1)
            if (v[i+j])
                add(i,j+LIM,1);
    while (bfs(0,H))
        mxFlow+=find(0,H,INF);
    fo(now,1,ans)
        if (lnk[now]){
            int i=now;
            do{
                if (i>LIM)
                    i-=LIM;
                cout<<i<<" ";
                int tmp=lnk[i];
                lnk[i]=0;
                i=tmp;
            }while (i);
            cout<<endl;
        }
    return 0;
}