bzoj 3158: 千钧一发

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

最近整个人都好懒。。还是见链接吧。

http://lgs-oi.blog.163.com/blog/static/231255002201452423021187/

#include<cmath>
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
int head[100001];
struct map
{
	 int f;
	 int s,t;
     int next;
}a[4000001];
int edge;
int p;
int q[400001],d[400001];
inline void add(int s,int t,int f)
{
     a[edge].next=head[s];
     head[s]=edge;
     a[edge].s=s;
     a[edge].t=t;
     a[edge].f=f;
}
inline bool bfs()
{
     int l=0,r=0;
     memset(q,0,sizeof(q));
     r++;
     q[r]=0;
     memset(d,-1,sizeof(d));
     d[0]=0;
     int i,k;
     while(l<r)
     {
     	  l++;
          int k=q[l];
          for(i=head[k];i!=0;i=a[i].next)
          {
               if(a[i].f>0&&d[a[i].t]==-1)
               {
                    d[a[i].t]=d[k]+1;
                    r++;
                    q[r]=a[i].t;
               }
          }
     }
     if(d[p]>=0)
          return true;
     return false;
}
inline int dfs(int k,int s)
{
     if(k==p)
          return s;
     int t=s;
     int i;
     for(i=head[k];i!=0;i=a[i].next)
     {
          if(d[a[i].t]==d[k]+1&&a[i].f>0)
          {
               int xx=dfs(a[i].t,min(s,a[i].f));
               a[i].f-=xx;
               if(i%2==0)
                    a[i-1].f+=xx;
               else
                    a[i+1].f+=xx;
               s-=xx;
          }
     }
     return t-s;
}
inline int maxflow()
{
     int s=0;
     while(bfs())
          s+=dfs(0,2100000000);
     return s;
}
inline int gcd(int x,int y)
{
     int m=x%y;
     while(m!=0)
     {
          x=y;
          y=m;
          m=x%y;
     }
     return y;
}
inline bool powed(int x,int y)
{
	 long long xx=(long long)x;
	 long long yy=(long long)y;
     long long t=xx*xx+yy*yy;
     long long tt=sqrt(t);
     if(tt*tt==t)
          return true;
     return false;
}
int number[1001];
int num[1001];
int main()
{
     int n;
     scanf("%d",&n);
     p=n+1;
     int i,j;
     int sum=0;
     for(i=1;i<=n;i++)
          scanf("%d",&number[i]);
     for(i=1;i<=n;i++)
     {
          scanf("%d",&num[i]);
          sum+=num[i];
          if(number[i]%2==0)
          {
               edge++;
               add(0,i,num[i]);
               edge++;
               add(i,0,0);
          }
          else
          {
               edge++;
               add(i,p,num[i]);
               edge++;
               add(p,i,0);
          }
     }
     for(i=1;i<=n;i++)
     {
     	  if(number[i]%2==1)
     	       continue;
          for(j=1;j<=n;j++)
          {
          	   if(i==j)
          	        continue;
               if(gcd(number[i],number[j])==1&&powed(number[i],number[j]))
               {
                    edge++;
                    add(i,j,2100000000);
                    edge++;
                    add(j,i,0);
               }
          }
     }
     printf("%d\n",sum-maxflow());
     return 0;
}