题目描述 Description
假设有 n 根柱子,现要按下述规则在这 n 根柱子中依次放入编号为 1,2,3,…的球。
(1)每次只能在某根柱子的最上面放球。
(2)在同一根柱子中,任何 2 个相邻球的编号之和为完全平方数。
试设计一个算法,计算出在 n 根柱子上最多能放多少个球。例如,在 4 根柱子上最多可放 11 个球。
对于给定的 n,计算在 n 根柱子上最多能放多少个球。
输入描述 Input Description
第 1 行有 1 个正整数 n,表示柱子数。
输出描述 Output Description
将 n 根柱子上最多能放的球数以及相应的放置方案输出
第一行是球数。接下来的 n 行,每行是一根柱子上的球的编号。
样例输入 Sample Input
4
样例输出 Sample Output
11
1 8
2 7 9
3 6 10
4 5 11
网络流这种题目看到说要输出详细结果的就很蛋疼,这种题目虽然最终结果是唯一的,但是过程又不唯一,要达到题目所要的输出结果代码估计唯一了。。
我在网上找了一下代码,发现这个题的最终结果可以用公式计算出来。
公式:输入一个n。(n^2-1)/2+n。公式来源:http://blog.csdn.net/delayyy/article/details/8605856
代码来源于:http://welcomechengyao.blog.163.com/blog/static/185469890201191523951283/
我对代码进行了略微修改,多了一个公式结果和几个汉字输出。。慢慢研究这个代码把。。哎。。。
以下是代码:
#include<iostream> #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> using namespace std; #define maxN 100000 #define maxE 200000 #define INF 1<<30 #define eps 0.000000001 struct Edge { int u,v,w,next; }; Edge edge[maxE]; int cnt=0; int cou=0; int s,t; int n; int K; int flow; int head[maxN],cur[maxN],pre[maxN],gap[maxN],dis[maxN]; int vis[maxN]; void addedge(int u,int v,int w) { edge[cnt].u=u; edge[cnt].v=v; edge[cnt].w=w; edge[cnt].next=head[u]; head[u]=cnt++; edge[cnt].u=v; edge[cnt].v=u; edge[cnt].w=0; edge[cnt].next=head[v]; head[v]=cnt++; } void sap() { for(int i=0; i<cou; i++) { cur[i]=head[i]; dis[i]=gap[i]=0; } gap[0]=cou; int u=pre[s]=s; int aug=INF; while(dis[s]<cou) { bool flag=true; for(int &j=cur[u]; j!=-1; j=edge[j].next) { int v=edge[j].v; if(edge[j].w>0 && dis[u]==dis[v]+1) { flag=false; pre[v]=u; u=v; if(aug>edge[j].w) aug=edge[j].w; if(u==t) { flow+=aug; while(u!=s) { u=pre[u]; edge[cur[u]].w-=aug; edge[cur[u]^1].w+=aug; } aug=INF; } break; } } if(!flag) continue; int minh=cou; for(int k=head[u]; k!=-1; k=edge[k].next) { int v=edge[k].v; if(edge[k].w>0 && minh>dis[v]) { minh=dis[v]; cur[u]=k; } } if((--gap[dis[u]])==0) break; gap[dis[u]=minh+1]++; u=pre[u]; } } bool check(int m,int n) { int sum=m+n; double dsum=sqrt(sum); int tem=(int)dsum; if(dsum-tem<eps) return true; else return false; } void dfs(int u,int clr) { vis[u]=clr; int v; bool flag=true; for(int i=head[u*2]; i!=-1; i=edge[i].next) { v=edge[i].v; if(edge[i].w==0 && v!=s) { flag=false; break; } } if(!flag) dfs(v/2,clr); } int main() { while(scanf("%d",&K)!=EOF) { //公式计算: int x; x = (K*K-1)/2+K; printf("公式计算结果:\n%d\n",x); memset(head,-1,sizeof(head)); cnt=0; cou=2; n=0; flow=0; s=0; t=1; while(n-flow<=K) { n++; cou+=2; addedge(s,n*2,1); addedge(n*2+1,t,1); for(int i=1; i<n; i++) { if(check(i,n)) addedge(i*2,n*2+1,1); } sap(); } n--; printf("网络流计算结果:\n%d\n详细结果如下:\n",n); memset(vis,-1,sizeof(vis)); int clr=0; for(int j=1; j<=n; j++) { if(vis[j]==-1) { dfs(j,clr++); } } for(int k=0; k<clr; k++) { int first=true; for(int j=1; j<=n; j++) { if(vis[j]==k) { if(first) { printf("%d",j); first=false; } else printf(" %d",j); } } printf("\n"); } } return 0; }