(这题很舒服。。。)
题目大意
题目写得很清楚。
假设有n根柱子,现要按下述规则在这n根柱子中依次放入编号为1,2,3,...的球。
(1)每次只能在某根柱子的最上面放球。
(2)在同一根柱子中,任何2个相邻球的编号之和为完全平方数。
试设计一个算法,计算出在n根柱子上最多能放多少个球。例如,在4 根柱子上最多可放11 个球。
«编程任务:
对于给定的n,计算在n根柱子上最多能放多少个球。
解法
没错,就是这样,是不是马上想到了暴力dfs!!!n只有55,但是你无法确定球的个数,所以dfs暴枚肯定会很慢(wjh:不然怎么出在网络流24题里)所以,我们要往网络流方面去想,而我们又要运用到网络流的反悔性质,所以很明显就会想到拆点.
我们每一次把一个球拆成两个点,前面的点连(源点),后面的点连(汇点),权值都为1,然后中间的怎么连呢?
很明显我们回记起来还有一个条件没有用上,那就是同一个柱子中,两个相邻的球的编号之和为完全平方数,所以如果a+b为完全平方数,那么我们就把a的前面那个店和b后面那个点连接起来,权值为一(注意:这里可能会理解错误,因为有些人可能想问x后面可以跟y,y后面可以跟z,那么是不是要把y的后面那个点与y前面那个点连起来,这样跑出来的结果才会对。。错!!!因为这个网络流中的流量只有两种状态,无解和有解,无解时跑出来的结果为0,有解时跑出来的结果>0,这样才能实现一次装一个柱子,要不你一次装多个柱子,那你怎么加点。。对吧!?)
我们每次加一个点,然后跑一次网络流,如果有解,那么就继续加点,否则,就将它压在第二个柱子的最底下,也就是里面的head[]数组。
说完了,看代码,喜欢我的点一波关注哦!
代码
#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
int n;//共有柱子
int spend=0;//开始用第几根柱子
int tot_ball=0;
int begin,end;
int head[60];//第i根柱子在柱底的是第几号球
struct edge{int y,next,c;};
edge s[400010];
int first[200010];
int match[50010];
int len=1;
int h[50010];
int f[100010];
void ins(int x,int y,int c)
{
len++;
s[len].y=y;s[len].c=c;
s[len].next=first[x];first[x]=len;
}
int min(int x,int y)
{
return x<y?x:y;
}
bool bfs()
{
int st=1,ed=2;
f[st]=begin;
memset(h,0,sizeof(h));
h[begin]=1;
while(st!=ed)
{
int x=f[st];
for(int i=first[x];i!=0;i=s[i].next)
{
int y=s[i].y;
if(h[y]==0 && s[i].c>0)
{
h[y]=h[x]+1;
f[ed]=y;
ed++;
if(ed==100000) ed=1;
}
}
st++;
if(st==100000) st=1;
}
return h[end];
}
int dfs(int x,int t)
{
if(x==end) return t;
int tot=0,flow=0;
for(int i=first[x];i!=0;i=s[i].next)
{
if(tot==t) return t;
int y=s[i].y;
if(s[i].c>0 && h[y]==h[x]+1)
{
flow=dfs(y,min(s[i].c,t-tot));
tot+=flow;s[i].c-=flow;s[i^1].c+=flow;
if(flow!=0 && y!=begin && y!=end && x%2==0)
match[x>>1]=y>>1;
}
}
if(tot==0) h[x]=0;
return tot;
}
int dinic()
{
int tot=0;
int dx;
while(bfs())
{
dx=dfs(begin,1e9);
while(dx!=0)
{
tot+=dx;
dx=dfs(begin,1e9);
}
}
return tot;
}
int main()
{
scanf("%d",&n);
begin=1;end=50000;
while(spend<=n)
{
tot_ball++;
ins(begin,tot_ball<<1,1);ins(tot_ball<<1,begin,0);
ins((tot_ball<<1)|1,end,1);ins(end,(tot_ball<<1)|1,0);
for(int i=sqrt(tot_ball)+1;i*i<2*tot_ball;i++)
{
ins((i*i-tot_ball)<<1,(tot_ball<<1)|1,1),ins((tot_ball<<1)|1,(i*i-tot_ball)<<1,0);
}
int k=dinic();
if(k==0) //因为没有流量流到汇点(也就是说,这个球只能放到下一个柱子了);
{//所以我们把它设为下一个柱子的开头
spend++;
head[spend]=tot_ball;
}
}
printf("%d\n",tot_ball-1);
bool visit[50010];//第i号球是否用过
memset(visit,false,sizeof(visit));
for(int i=1;i<=n;i++)
{
if(visit[head[i]]) continue;
int now=head[i];
while(now!=0 && visit[now]==false)
{
printf("%d ",now);
visit[now]=true;
now=match[now];
}
printf("\n");
}
}