NKOI 1939 魔术球

【线性规划与网络流24题 4】魔术球

Time Limit:1000MS  Memory Limit:65536K
Total Submit:22 Accepted:7

Description

假设有n根柱子，现要按下述规则在这n根柱子中依次放入编号为 1，2，3，...的球。 
（1）每次只能在某根柱子的最上面放球。 
（2）在同一根柱子中，任何2个相邻球的编号之和为完全平方数。 
试设计一个算法，计算出在n根柱子上最多能放多少个球。例如，在4 根柱子上最多可放11个球。 
编程任务： 对于给定的n，计算在 n根柱子上最多能放多少个球。 

由于本OJ无Special Judge ， 所以只需要输出最多能放的球的个数

Input

第1 行有 1个正整数n，表示柱子数。(0<n<60)

Output

一行，包含一个整数，表示最多能放的球数

Sample Input

4

Sample Output

11

Source

感谢 Wo_ai_WangYuan 修改题目并放上数据

一个裸的最小路径覆盖

将每一个对i,j(i<j)进行讨论，如果i+j是完全平方数，就将i和j连一条容量为1的边

在图中求到的最大流实际上就是最大匹配的对数，减去这个数得到的就是能够"覆盖"的柱子的个数

因此答案需要枚举，可以顺序枚举也可以二分，效果是一样的，都会超时。

用sap怎么貌似都要超时，因此写了一个效率相对比较高的，打了个表……

把表放出来就没意思了，因此放出来自己写的最快的一个，但是还是超时……

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=50000,inf=1e9;
int n,dis[maxn],vd[maxn],cnt=-1,ans=0,op,ed;
int END[maxn],last[maxn],NEXT[maxn],len[maxn];
void insert(int a,int b){
	END[++cnt]=b;
	len[cnt]=1;
	NEXT[cnt]=last[a];
	last[a]=cnt;
	
	END[++cnt]=a;
	len[cnt]=0;
	NEXT[cnt]=last[b];
	last[b]=cnt;
}
int dfs(int u,int flow){    
    if(u==ed)return flow;    
    int i,temp,delta=0;    
    for(i=last[u];i>=0;i=NEXT[i]){
    	int v=END[i];
    	if(len[i]&&dis[u]==dis[v]+1){
    		temp=dfs(v,min(flow-delta,len[i]));
    		len[i]-=temp;
    		len[i^1]+=temp;
            delta+=temp;    
            if(delta==flow||dis[0]>=ed)return delta; 
		}
	}  
    if(dis[0]>ed)return delta;    
    vd[dis[u]]--;    
    if(vd[dis[u]]==0)dis[0]=ed;    
    dis[u]++;    
    vd[dis[u]]++;    
    return delta;    
}
void clear(){
	memset(dis,0,sizeof(dis));
	memset(vd,0,sizeof(vd));
	memset(last,-1,sizeof(last));
	memset(NEXT,-1,sizeof(NEXT));
	memset(len,0,sizeof(len));
	memset(END,-1,sizeof(END));
	cnt=-1;
}
bool check(int tg){
	int ans=0,i,j;
	clear();
	op=tg*2+1,ed=op+1;
	for(i=1;i<tg;i++){
		insert(op,i);
		insert(i+tg,ed);
	    for(j=i+1;j<=tg;j++)
		    if(sqrt(i+j)==(int)sqrt(i+j))
	            insert(i,j+tg);
	}
	insert(op,tg);insert(tg+tg,ed);
	while(dis[op]<ed)ans+=dfs(op,inf);
	if(tg-ans<=n)return 0;
	return 1;
}
int main(){
	cin>>n;
	int l=1,r=1805;
	while(l<=r){
		int mid=(l+r)/2;
		if(check(mid))r=mid-1;
		else l=mid+1;
	}
	cout<<l-1<<endl;
}