【网络流24题】魔术球问题

Description

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

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

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

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

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

Input

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

Output

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

Sample Input

4

Sample Output

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

HINT

n<60

题解

网络流最大流
枚举答案s，在图中建节点1,2,….,s，并将每个节点拆成两个点，一个连接原点，一个连接汇点，对于每一个i< j，如果存在i+j是一个平方数，那么就将i与原点连接的点和j与汇点连接的点建边。每次求一遍最大流a，如果a-1==n，那么s-1即为所求解。

code

#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include