3158: 千钧一发 最小割

我收回我在3275: Number 最小割中说的话……
奇偶建图才是正确的姿势，暴力会T的很惨。。
我们可以发现，两个奇数一定满足1，而两个偶数一定满足2，所以我们可以奇偶建图，如果一个奇数和一个偶数不能同时选，那么就连边inf，奇偶分别连向两边，跑最小割。

10^6的平方是会爆longlong的。。。QAQ

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define inf 2147483647
#define ll long long 
using namespace std;
int n,sum,cnt=1,ans,T;
int a[1005],b[1005],head[2005],cur[2005],dis[2005],q[2005];
int list[1000005],next[1000005],key[1000005];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void insert(int x,int y,int z)
{
    next[++cnt]=head[x];
    head[x]=cnt;
    list[cnt]=y;
    key[cnt]=z;
}
int gcd(int a,int b)
{
    return b==0?a:gcd(b,a%b);
}
inline bool judge(int i,int j)
{
    ll c=(ll)a[i]*a[i]+(ll)a[j]*a[j],d=(ll)(sqrt(c));
    if (d*d!=c) return 0;
    if (gcd(a[i],a[j])!=1) return 0;
    return 1;
}
inline bool BFS()
{
    memset(dis,-1,sizeof(dis));
    dis[0]=1; q[1]=0;
    int t=0,w=1,x;
    while (t<w)
    {
        x=q[++t];
        for (int i=head[x];i;i=next[i])
            if (key[i]&&dis[list[i]]==-1)
                dis[list[i]]=dis[x]+1,q[++w]=list[i];
    }
    return dis[T]!=-1;
}
int find(int x,int flow)
{
    if (x==T) return flow;
    int w,used=0;
    for (int i=cur[x];i;i=next[i])
        if (key[i]&&dis[list[i]]==dis[x]+1)
        {
            w=find(list[i],min(key[i],flow-used));
            key[i]-=w; key[i^1]+=w; used+=w;
            if (key[i]) cur[x]=i;
            if (used==flow) return flow;
        }
    if (!used) dis[x]=-1;
    return used;
}
int main()
{
    n=read(); T=2*n+1;
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++) b[i]=read(),sum+=b[i];
    for (int i=1;i<=n;i++)
        if (a[i]&1) insert(0,i,b[i]),insert(i,0,0);
        else insert(i,T,b[i]),insert(T,i,0);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            if ((a[i]&1)&&!(a[j]&1))
                if (judge(i,j)) insert(i,j,inf),insert(j,i,0);
    while (BFS()) 
    {
        for (int i=0;i<=T;i++) cur[i]=head[i];
        ans+=find(0,inf);
    }
    cout << sum-ans;
    return 0;
}