【BZOJ3158】千钧一发，网络流之最大权闭合子图

传送门
写在前面：学弟好可怕
思路：
有一个引理，如果x,y,z互质且x^2+y^2=z^2，那么x，y必定一奇一偶。（详见百度百科）
有了这个就好办了，把特征值a[i]分成奇数偶数两部分，s连奇数，偶数连t，容量为b[i]，符合条件的两个数之间连一条容量为inf的边，最大权闭合子图跑出来的最大流就是要舍弃的数的b[i]和，用∑b[i]减去就可以了
注意：naive的我以为是奇数偶数一定互质……

#include<bits/stdc++.h>
#define LL long long
#define inf 0x7fffff
using namespace std;
int n,tot=1,s,t,ans;
int a[3010],b[3010],first[3010],dis[3010],One[3010],Two[3010];
struct edge
{
    int u,v,w,next;
}e[2050000];
void add(int x,int y,int z){e[++tot]=(edge){x,y,z,first[x]};first[x]=tot;};
queue<int>q;
bool pd(LL x)
{
    LL t=sqrt(x); 
    return t*t==x;
}
int gcd(int x,int y)
{
    if (!y) return x;
    return gcd(y,x%y);
} 
bool bfs()
{
    memset(dis,0,sizeof(dis));
    dis[s]=1;q.push(s);
    while (!q.empty())
    {
        int k=q.front();
        q.pop();
        for (int i=first[k];i;i=e[i].next)
            if (!dis[e[i].v]&&e[i].w)
                q.push(e[i].v),dis[e[i].v]=dis[k]+1;
    }
    return dis[t];
}
int dfs(int x,int maxn)
{
    if (x==t) return maxn;
    int used=0;
    for (int i=first[x];i;i=e[i].next)
        if (dis[e[i].v]==dis[x]+1)
        {
            int k=dfs(e[i].v,min(maxn-used,e[i].w));
            e[i].w-=k;e[i^1].w+=k;
            used+=k;
            if (used==maxn) return maxn;
        }
    if (!used) dis[x]=0;
    return used;
}
main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",a+i);
    t=n+1;
    for (int i=1;i<=n;i++)
    {
        scanf("%d",b+i); 
        ans+=b[i];
        if (a[i]&1) One[++One[0]]=i,add(s,i,b[i]),add(i,s,0);
        else Two[++Two[0]]=i,add(i,t,b[i]),add(t,i,0);
    }
    for (int i=1;i<=One[0];i++)
        for (int j=1;j<=Two[0];j++)
        if (gcd(a[One[i]],a[Two[j]])==1&&pd((LL)a[One[i]]*a[One[i]]+(LL)a[Two[j]]*a[Two[j]]))
            add(One[i],Two[j],inf),
            add(Two[j],One[i],0);
    while (bfs()) ans-=dfs(s,inf);
    printf("%d",ans);
}