「BZOJ3158」千钧一发
Description
Input
第一行一个正整数N。
第二行共包括N个正整数,第 个正整数表示Ai。
第三行共包括N个正整数,第 个正整数表示Bi。
Output
共一行,包括一个正整数,表示在合法的选择条件下,可以获得的能量值总和的最大值。
Sample Input
4
3 4 5 12
9 8 30 9
Sample Output
39
HINT
1<=N<=1000,1<=Ai,Bi<=10^6
题解
可以证明,任意两个偶数满足2
两个奇数满足1
(2a+1)^2+(2b+1)^2=2(2a^2+2b^2+2a+2b+1)
一定不是完全平方数
所以可以用最小割解决此题
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
#include<iostream> #include<cstdio> #include<cstdlib> #include<algorithm> #include<cmath> #include<cstring> #define ll long long #define inf 2000000000 using namespace std; inline int read() { int x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } 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() { n=read();T=n+1; for(int i=1;i<=n;i++)a[i]=read(); for(int i=1;i<=n;i++)b[i]=read(),ans+=b[i]; build(); dinic(); printf("%d",ans); return 0; } |
第一眼最大团给跪了.