【BZOJ3275】Number 最小割
【BZOJ3275】Number
Description
有N个正整数,需要从中选出一些数,使这些数的和最大。
若两个数a,b同时满足以下条件,则a,b不能同时被选
1:存在正整数C,使a*a+b*b=c*c
2:gcd(a,b)=1
Input
第一行一个正整数n,表示数的个数。
第二行n个正整数a1,a2,?an。
Output
最大的和。
Sample Input
5
3 4 5 6 7
3 4 5 6 7
Sample Output
22
HINT
n<=3000。
题解:先是无脑码了个最小割,果断WA了,看网上才又get了一个新定理
易证:奇数和奇数的平方和一定不是完全平方数,偶数和偶数的gcd一定不为1
然后就把所有的数分成奇数和偶数两个集合,然后再跑最小割就完事了
#include <cstdio> #include <cstring> #include <iostream> #include <queue> #include <algorithm> #include <cmath> using namespace std; int n,cnt,tot,ans,tx,ty; queue<int> q; int vx[3010],vy[3010],next[1000000],head[6010],to[1000000],val[1000000],d[6010]; int gcd(int a,int b) { return (b==0)?a:gcd(b,a%b); } int dfs(int x,int mf) { if(x==n+1) return mf; int i,k,temp=mf; for(i=head[x];i!=-1;i=next[i]) { if(d[to[i]]==d[x]+1&&val[i]) { k=dfs(to[i],min(temp,val[i])); if(!k) d[to[i]]=0; val[i]-=k,val[i^1]+=k,temp-=k; if(!temp) break; } } return mf-temp; } int bfs() { int i,u; memset(d,0,sizeof(d)); while(!q.empty()) q.pop(); q.push(0),d[0]=1; while(!q.empty()) { u=q.front(),q.pop(); for(i=head[u];i!=-1;i=next[i]) { if(!d[to[i]]&&val[i]) { d[to[i]]=d[u]+1; if(to[i]==n+1) return 1; q.push(to[i]); } } } return 0; } void add(int a,int b,int c) { to[cnt]=b; val[cnt]=c; next[cnt]=head[a]; head[a]=cnt++; } int main() { scanf("%d",&n); int i,j,k; memset(head,-1,sizeof(head)); for(i=1;i<=n;i++) { scanf("%d",&k); tot+=k; if(k&1) vx[++tx]=k; else vy[++ty]=k; } for(i=1;i<=tx;i++) add(0,i,vx[i]),add(i,0,0); for(i=1;i<=ty;i++) add(i+tx,n+1,vy[i]),add(n+1,i+tx,0); for(i=1;i<=tx;i++) { for(j=1;j<=ty;j++) { if(gcd(vx[i],vy[j])!=1) continue; int k=vx[i]*vx[i]+vy[j]*vy[j]; if(int(sqrt(k*1.0)+0.00001)*int(sqrt(k*1.0)+0.00001)==k) { add(i,tx+j,1<<30); add(tx+j,i,0); } } } while(bfs()) ans+=dfs(0,1<<30); printf("%d",tot-ans); return 0; }
| 欢迎来原网站坐坐! >原文链接<