网络流24题 魔术球问题(弱化？)(二分图解)

经典的网络流问题

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

4

Sample Output

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

　一眼思路：搜索每个数放的位置判断解。

　　　　　显然会T到死，于是转换下思路(毕竟图论)。

　转换思路：

　　　　　枚举球的总数s，因为球的数量多了之后一定需要放开的柱子数也会增多(即符合单调)，自底向上枚举球的数量s后计算出需要的最少的柱子数在第一个比给出的n大的s是第一个不可行解，s - 1即是答案。

　……那么问题来了，如何计算s个球所需的最少柱子数。

　这个过程我们可以看做首先把s个球分别放在s个柱子上，这是最大的柱子数

　然后把两个球放在一起，假如说是i和j，那么就保证i + j是完全平方数且i < j，也就是合并了两个球。

　每次合并所需柱子数都会减一，所以就要找最大的合并数。

　把每个球拆成两个，在左手边(?)放一个右手边放一个，对于左边的每个i找右边的一个满足( j > i && (i + j) == 完全平方数 )的j，连一条边，表示这两个可以进行合并。

　最后左手边的球和右手边的球构成一张二分图，在这张二分图上跑下最大匹配得出最大合并次数ans，用s - ans就是最少用的柱子数。

　我们来顺下思路，枚举答案s，计算最少用的柱子数，如果大于n，则输出s - 1这就是第一问的答案。

　那么第二问呢？说真的没做(滑稽)，不过我觉得这样跑出来的二分图只要把每个i拎出来把和它相连的依次取出，输出即可。第二问是O(n)的。

　二分图可以用网络流做嘛，所以网络流肯定可以的。

代码

#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int Maxn = 2000;

vector <int> q[Maxn*2+10];

int n,Link[Maxn*2+10];
bool vis[Maxn*2+10];

bool judge(int x){
    int y = sqrt(x);
    if(y * y == x)return 1;
    else return 0;  
}

int Work(int x){
    int T = q[x].size();
    for(int i=0;i<T;i++){
        int v = q[x][i];
        if(!vis[v]){
            vis[v] = 1;
            if(!Link[v] || Work(Link[v])){
                Link[v] = x;
                return 1;
            }
        }
    }
    return 0;
}

int main(){
//  freopen("balla.in","r",stdin);
//  freopen("balla.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=2001;i++){
        int ans = 0;
        for(int j=1;j<i;j++)if(judge(i + j))q[j].push_back(i + Maxn);
        for(int k=1;k<i;k++)Link[k+2000] = 0;
        for(int k=1;k<=i;k++){
            memset(vis,0,sizeof(vis));
            if(Work(k))ans++;
        }
        if(i - ans > n){
            printf("%d\n",i - 1);
            return 0;
        }
    }return 0;
}

我只做了第一问23333