记录编号 304418 评测结果 AAAAAAAAAA
题目名称 [网络流24题]魔术球问题(简化版) 最终得分 100
用户昵称 GravatarAntiLeaf 是否通过 通过
代码语言 C++ 运行时间 2.889 s
提交时间 2016-09-08 14:16:17 内存使用 12.46 MiB
显示代码纯文本
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=1610;
struct edge{
	int to,prev;
}e[2000010];
void insert(int,int);
bool findpath(int,int);
int Hungary();
int last[maxn],ca[maxn],cb[maxn];
bool vis[maxn];
int n,l=0,r=1600,ans,len=0;
int main(){
#define MINE
#ifdef MINE
	freopen("balla.in","r",stdin);
	freopen("balla.out","w",stdout);
#endif
	scanf("%d",&n);
	while(l<=r){
		ans=(l+r)>>1;
		memset(last,0,sizeof(last));
		len=0;
		for(int i=1;i<=ans;i++)for(int j=1;j<i;j++)
			if((int)(sqrt(i+j)+1e-5)*(int)(sqrt(i+j)+1e-5)==i+j)insert(j,i);
		if(ans-Hungary()<=n)l=ans+1;
		else r=ans-1;
	}
	printf("%d",r);
#ifndef MINE
	printf("\n--------------------DONE--------------------\n");
	for(;;);
#endif
	return 0;
}
void insert(int x,int y){
	e[++len].to=y;
	e[len].prev=last[x];
	last[x]=len;
}
bool findpath(int x){
#define y e[i].to
	for(int i=last[x];i;i=e[i].prev)if(!vis[e[i].to]){
		vis[e[i].to]=true;
		if(!cb[e[i].to]||findpath(cb[e[i].to])){
			ca[x]=e[i].to;
			cb[e[i].to]=x;
			return true;
		}
	}
	return false;
}
int Hungary(){
	int res=0;
	memset(ca,0,sizeof(ca));
	memset(cb,0,sizeof(cb));
	for(int i=1;i<=ans;i++)if(!ca[i]){
		memset(vis,0,sizeof(vis));
		if(findpath(i))res++;
	}
	return res;
}