bzoj3275: Number

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cmath>
 4 #include<cstring>
 5 #include<algorithm>
 6 #define maxn 3005
 7 #define maxm 6000000
 8 #define inf 1061109567
 9 using namespace std;
10 char ch;
11 bool ok;
12 void read(int &x){
13     for (ok=0,ch=getchar();!isdigit(ch);ch=getchar()) if (ch==‘-‘) ok=1;
14     for (x=0;isdigit(ch);x=x*10+ch-‘0‘,ch=getchar());
15     if (ok) x=-x;
16 }
17 int n,a[maxn],sum;
18 struct flow{
19     int s,t,tot,now[maxn],son[maxm],pre[maxm],val[maxm];
20     int dis[maxn],head,tail,list[maxn];
21     bool bo[maxn];
22     void init(){s=0,t=n+1,tot=1,memset(now,0,sizeof(now));}
23     void put(int a,int b,int c){pre[++tot]=now[a],now[a]=tot,son[tot]=b,val[tot]=c;}
24     void add(int a,int b,int c){put(a,b,c),put(b,a,0);}
25     bool bfs(){
26         memset(bo,0,sizeof(bo));
27         head=0,tail=1,list[1]=s,dis[s]=0,bo[s]=1;
28         while (head<tail){
29             int u=list[++head];
30             for (int p=now[u],v=son[p];p;p=pre[p],v=son[p])
31                 if (val[p]&&!bo[v]) bo[v]=1,dis[v]=dis[u]+1,list[++tail]=v;
32         }
33         return bo[t];
34     }
35     int dfs(int u,int rest){
36         if (u==t) return rest;
37         int ans=0;
38         for (int p=now[u],v=son[p];p&&rest;p=pre[p],v=son[p])
39             if (val[p]&&dis[v]==dis[u]+1){
40                 int d=dfs(v,min(rest,val[p]));
41                 val[p]-=d,val[p^1]+=d,ans+=d,rest-=d;
42             }
43         if (!ans) dis[u]=-1;
44         return ans;
45     }
46     int dinic(){
47         int ans=0;
48         while (bfs()) ans+=dfs(s,inf);
49         return ans;
50     }
51 }f;
52 int gcd(int a,int b){return b?gcd(b,a%b):a;}
53 int main(){
54     read(n),f.init();
55     for (int i=1;i<=n;i++){
56         read(a[i]),sum+=a[i];
57         if (a[i]&1) f.add(i,f.t,a[i]); else f.add(f.s,i,a[i]);
58     }
59     for (int i=1;i<=n;i++) if (!(a[i]&1)) for (int j=1;j<=n;j++) if (a[j]&1){
60         int tmp=round(sqrt(a[i]*a[i]+a[j]*a[j]));
61         if (gcd(a[i],a[j])==1&&tmp*tmp==a[i]*a[i]+a[j]*a[j]) f.add(i,j,inf);
62     }
63     printf("%d\n",sum-f.dinic());
64     return 0;
65 }