「网络流 24 题」魔术球

题目描述

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

1. 每次只能在某根柱子的最上面放球。
2. 在同一根柱子中，任何 $2$ 个相邻球的编号之和为完全平方数。

试设计一个算法，计算出在 $n$ 根柱子上最多能放多少个球。

输入格式

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

输出格式

第一行是球数。接下来的 $n$ 行，每行是一根柱子上的球的编号。

样例输入

4

样例输出

11

1 8

2 7 9

3 6 10

4 5 11

题目要求每个点只使用一次且连续，如果把没条柱子上的点集看成一条路径，那么就相当于求一个最大ans使得1~ans点集构成的最小路径覆盖>=n了。建图类似一个二分匹配，将每个点拆成一个入点和一个出点，对于每一个i，如果i<j且i+j是完全平方数，则将i的入点与j的出点相连，源点和

每个点的入点相连，每个点的出点和汇点相连，从小到大枚举ans直到不满足情况位置，答案就是ans-1。注意这里并不超时，每次枚举都是在之前的残留路径上继续增广。输出路径需要注意一下。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<queue>
using namespace std;
const int maxm = 100005;
const int INF = 1e9 + 7;
int flag[maxm], dis[maxm], vis[maxm], head[maxm], cur[maxm], to[maxm];
int cnt, n, m, s, t, len, ans;
struct node
{
	int v, flow, next;
}edge[maxm];
void init()
{
	cnt = 0, s = 0, t = 10005, m = 5000, ans = 0;
	memset(head, -1, sizeof(head));
}
void add(int u, int v, int w)
{
	edge[cnt].v = v, edge[cnt].flow = w, edge[cnt].next = head[u], head[u] = cnt++;
	edge[cnt].v = u, edge[cnt].flow = 0, edge[cnt].next = head[v], head[v] = cnt++;
}
int bfs()
{
	queue<int>q;
	memset(dis, -1, sizeof(dis));
	dis[s] = 0;
	q.push(s);
	int rev = 0;
	while (!q.empty())
	{
		int u = q.front();q.pop();
		for (int i = head[u];i != -1;i = edge[i].next)
		{
			int v=edge[i].v;
			if (dis[v] == -1 && edge[i].flow)
			{
				dis[v]=dis[u]+1;
				q.push(v);
			}
		}
	}
	if (dis[t] == -1) return 0;
	return 1;
}
int dfs(int u, int flow)
{
	if (u == t) return flow;
	for (int i = cur[u];i != -1;i = edge[i].next)
	{
		int v = edge[i].v;
		if (dis[v] == dis[u] + 1 && edge[i].flow)
		{
			int d = dfs(v, min(edge[i].flow, flow));
			if (d > 0)
			{
				edge[i].flow -= d, edge[i ^ 1].flow += d;
				return d;
			}
		}
	}
	return 0;
}
void dinic()
{
	int d;
	while (bfs())
	{
		for (int i = 0;i <= t;i++) cur[i] = head[i];
		while (d = dfs(s, INF))
			ans += d;
	}
}
int main()
{
	int i, j;
	scanf("%d", &n);
	len = 0;init();
	while (1)
	{
		len++;
		add(s, len, 1), add(len + m, t, 1);
		for (i = 1;i < len;i++)
			if (sqrt(i + len) == (int)sqrt(i + len))
				add(i, len + m, 1);
		dinic();
		if (len - ans > n) break;
	}
	printf("%d\n", len - 1);
	memset(vis, 0, sizeof(vis));
	for (int i = 1;i < len;i++)
	{
		for (int j = head[i];j != -1;j = edge[j].next)
		{
			if (edge[j].flow)continue;
			to[i] = edge[j].v - m;
			break;
		}
	}
	for (int i = 1;i < len;i++)
	{
		if (vis[i])continue;
		int now = i;
		while (now != -m)
		{
			printf("%d ", now);
			vis[now] = true;
			now = to[now];
		}
		printf("\n");
	}
}