Description
假设有n根柱子,现要按下述规则在这n根柱子中依次放入编号为1,2,3,…的球。
每次只能在某根柱子的最上面放球。
在同一根柱子中,任何2个相邻球的编号之和为完全平方数。
试设计一个算法,计算出在n根柱子上最多能放多少个球。例如,在4 根柱子上最多可放11 个球。
Analysis
经过再三地读题之后可以发现
- 球的数量是未知的
- 随着球的数量增加,所需要的柱子是非递减的
这些就够了,那么我们可以考虑枚举球的个数然后对编号和为平方数的球连边,求出的最小路径覆盖就是柱子数量所以找到第一个路径覆盖大于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;
}