网络流/最小割

　　这题跟BZOJ 3275限制条件是一样的= =所以可以用相同的方法去做……只要把边的容量从a[i]改成b[i]就行了～

（果然不加当前弧优化要略快一点）

 /**************************************************************

     Problem: 3158

     User: Tunix

     Language: C++

     Result: Accepted

     Time:196 ms

     Memory:13028 kb

 ****************************************************************/

 //BZOJ 3158

 #include<cmath>

 #include<vector>

 #include<cstdio>

 #include<cstring>

 #include<cstdlib>

 #include<iostream>

 #include<algorithm>

 #define rep(i,n) for(int i=0;i<n;++i)

 #define F(i,j,n) for(int i=j;i<=n;++i)

 #define D(i,j,n) for(int i=j;i>=n;--i)

 #define pb push_back

 using namespace std;

 inline int getint(){

     int v=,sign=; char ch=getchar();

     while(ch<''||ch>''){ if (ch=='-') sign=-; ch=getchar();}

     while(ch>=''&&ch<=''){ v=v*+ch-''; ch=getchar();}

     return v*sign;

 }

 const int N=,M=,INF=~0u>>;

 typedef long long LL;

 /******************tamplate*********************/

 int n,m,tot,ans,a[N],b[N];

 struct edge{int to,v;};

 int gcd(int a,int b){return b ? gcd(b,a%b) : a;}

 bool judge(LL a,LL b){

     LL s=a*a+b*b;

     LL q=sqrt(s);

     if (q*q!=s) return ;

     if (gcd(a,b)!=) return ;

     return ;

 }

 struct Net{

     edge E[M];

     int head[N],next[M],cnt;

     void ins(int x,int y,int v){

         E[++cnt]=(edge){y,v};

         next[cnt]=head[x]; head[x]=cnt;

     }

     void add(int x,int y,int v){

         ins(x,y,v); ins(y,x,);

     }

     int s,t,d[N],Q[N];

     void init(){

         n=getint(); cnt=;

         tot=ans=;

         s=; t=n+;

         F(i,,n) a[i]=getint();

         F(i,,n){

             b[i]=getint();tot+=b[i];

             if (a[i]&) add(s,i,b[i]);

             else add(i,t,b[i]);

         }

         F(i,,n) if (a[i]&)

             F(j,,n) if((a[j]&)==)

                 if (judge(a[i],a[j])) add(i,j,INF);

     }

     bool mklevel(){

         memset(d,-,sizeof d);

         d[s]=;

         int l=,r=-;

         Q[++r]=s;

         while(l<=r){

             int x=Q[l++];

             for(int i=head[x];i;i=next[i])

                 if (d[E[i].to]==- && E[i].v){

                     d[E[i].to]=d[x]+;

                     Q[++r]=E[i].to;

                 }

         }

         return d[t]!=-;

     }

     int dfs(int x,int a){

         if (x==t) return a;

         int flow=;

         for(int i=head[x];i && flow<a;i=next[i])

             if (E[i].v && d[E[i].to]==d[x]+){

                 int f=dfs(E[i].to,min(a-flow,E[i].v));

                 E[i].v-=f;

                 E[i^].v+=f;

                 flow+=f;

             }

         if (!flow) d[x]=-;

         return flow;

     }

     void Dinic(){

         while(mklevel())

             ans+=dfs(s,INF);

     }

 }G1;

 int main(){

 #ifndef ONLINE_JUDGE

     freopen("3158.in","r",stdin);

     freopen("3158.out","w",stdout);

 #endif

     G1.init(); G1.Dinic();

     printf("%d\n",tot-ans);

     return ;

 }

3158: 千钧一发

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 590 Solved: 225
[Submit][Status][Discuss]

Description

Input

第一行一个正整数N。

第二行共包括N个正整数，第个正整数表示Ai。

第三行共包括N个正整数，第个正整数表示Bi。

Output

共一行，包括一个正整数，表示在合法的选择条件下，可以获得的能量值总和的最大值。

Sample Input

4
3 4 5 12
9 8 30 9

Sample Output

HINT

1<=N<=1000,1<=Ai,Bi<=10^6

Source

Katharon+#1

[Submit][Status][Discuss]

秒客网

【BZOJ】【3158】千钧一发