3158: 千钧一发

3158: 千钧一发

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1510  Solved: 556
[ Submit][ Status][ Discuss]

Description

Input

第一行一个正整数N。

第二行共包括N个正整数，第 个正整数表示Ai。

第三行共包括N个正整数，第 个正整数表示Bi。

Output

共一行，包括一个正整数，表示在合法的选择条件下，可以获得的能量值总和的最大值。

Sample Input

4
3 4 5 12
9 8 30 9

Sample Output

39

题解：

一道有意思的题。

易得奇数都满足1，偶数都满足2

所以，我们可以·把st连奇数，偶数连ed。

然后手动判断，奇数和偶数可不可以连。如果不可以的话，就建一条流量为inf的边。

#include
#include
#include
#include
#include
#include
#define ll long long 
#define inf 0x7fffffff
using namespace std;
int n,cnt=1,T,ans;
int a[1005],b[1005],last[1005],cur[1005];
int h[1005],q[1005];
struct edge{
	int to,next,v;
}e[1000005];
int gcd(int a,int b)
{
	return b==0?a:gcd(b,a%b);
}
void insert(int u,int v,int w)
{
	e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;e[cnt].v=w;
	e[++cnt].to=u;e[cnt].next=last[v];last[v]=cnt;e[cnt].v=0;
}
bool jud(ll x,ll y)
{
	ll t=x*x+y*y,sq=sqrt(t);
	if(sq*sq!=t)return 1;
	if(gcd(x,y)>1)return 1;
	return 0;
}
void build()
{
	for(int i=1;i<=n;i++)
		if(a[i]%2==1)insert(0,i,b[i]);
		else insert(i,T,b[i]);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			if((a[i]%2==1)&&(a[j]%2==0))
				if(!jud(a[i],a[j]))insert(i,j,inf);
}
bool bfs()
{
	int head=0,tail=1;
	for(int i=0;i<=T;i++)h[i]=-1;
	q[0]=0;h[0]=0;
	while(head!=tail)
	{
		int now=q[head];head++;
		for(int i=last[now];i;i=e[i].next)
			if(e[i].v&&h[e[i].to]==-1)
			{
				h[e[i].to]=h[now]+1;
				q[tail++]=e[i].to;
			}
	}
	return h[T]!=-1;
}
int dfs(int x,int f)
{
	if(x==T)return f;
	int w,used=0;
	for(int i=cur[x];i;i=e[i].next)
		if(h[e[i].to]==h[x]+1)
		{
			w=dfs(e[i].to,min(f-used,e[i].v));
			e[i].v-=w;e[i^1].v+=w;
			if(e[i].v)cur[x]=i;
			used+=w;if(used==f)return f;
		}
	if(!used)h[x]=-1;
	return used;
}
void dinic()
{
	while(bfs())
	{
		for(int i=0;i<=T;i++)
			cur[i]=last[i];
		ans-=dfs(0,inf);
	}
}
int main()
{
	scanf("%d",&n);T=n+1;
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)scanf("%d",&b[i]),ans+=b[i];
	build();
	dinic();
	printf("%d",ans);
	return 0;
}