BZOJ 3158: 千钧一发

3158: 千钧一发

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1201  Solved: 446
[Submit][Status][Discuss]

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

Source

Katharon+#1

[Submit][Status][Discuss]

网络流求最小割，很机智的建图方式。

发现可以把数字按照奇偶性分类，奇数一定满足1条件，偶数一定满足2条件，WOC，然后就二分图了？

暴力判断奇数和偶数是否不能在同一集合中，如不能在同一集合，在其中加一天正无穷的边，表示不可割。

其余点按奇偶性分别向S，T连bi的边即可，Σbi - 最小割就是最终答案。

 #include <cmath>
 #include <cstdio>
 #include <cstring>

 typedef long long lnt;

 const int siz = ;
 const int inf = ;

 int n;
 int a[siz];
 int b[siz];

 int tot;
 int s, t;
 int hd[siz];
 int to[siz];
 int fl[siz];
 int nt[siz];

 inline void add(int u, int v, int f)
 {
     nt[tot] = hd[u]; to[tot] = v; fl[tot] = f; hd[u] = tot++;
     nt[tot] = hd[v]; to[tot] = u; fl[tot] = ; hd[v] = tot++;
 }

 int dep[siz];

 inline bool bfs(void)
 {
     static int que[siz];
     static int head, tail;

     memset(dep, , sizeof(dep));
     head = , tail = ;
     que[tail++] = s;
     dep[s] = ;

     while (head != tail)
     {
         int u = que[head++], v;

         for (int i = hd[u]; ~i; i = nt[i])
             if (!dep[v = to[i]] && fl[i])
                 dep[que[tail++] = v] = dep[u] + ;
     }

     return dep[t];
 }

 int cur[siz];

 int min(int a, int b)
 {
     return a < b ? a : b;
 }

 int dfs(int u, int f)
 {
     if (u == t || !f)
         return f;

     int used = , flow, v;

     for (int i = cur[u]; ~i; i = nt[i])
         if (dep[v = to[i]] == dep[u] +  && fl[i])
         {
             flow = dfs(v, min(f - used, fl[i]));

             used += flow;
             fl[i] -= flow;
             fl[i^] += flow;

             if (used == f)
                 return f;

             if (fl[i])
                 cur[u] = i;
         }

     if (!used)
         dep[u] = ;

     return used;
 }

 inline int maxFlow(void)
 {
     int maxFlow = , newFlow;

     while (bfs())
     {
         for (int i = s; i <= t; ++i)
             cur[i] = hd[i];

         while (newFlow = dfs(s, inf))
             maxFlow += newFlow;
     }

     return maxFlow;
 }

 inline lnt sqr(lnt x)
 {
     return x * x;
 }

 int gcd(int x, int y)
 {
     return y ? gcd(y, x % y) : x;
 }

 inline bool check(int x, int y)
 {
     if (gcd(x, y) != )
         return false;

     lnt t = sqr(x) + sqr(y);
     if (sqr(sqrt(t)) != t)
         return false;

     return true;
 }

 signed main(void)
 {
     scanf("%d", &n);

     for (int i = ; i <= n; ++i)
         scanf("%d", a + i);

     for (int i = ; i <= n; ++i)
         scanf("%d", b + i);

     s = , t = n + ;

     memset(hd, -, sizeof(hd));

     for (int i = ; i <= n; ++i)
         if (a[i] & )
             add(s, i, b[i]);
         else
             add(i, t, b[i]);

     for (int i = ; i <= n; ++i)
         for (int j = ; j <= n; ++j)
             if (a[i] & )if (!(a[j] & ))
                 if (check(a[i], a[j]))
                     add(i, j, inf);

     int sum = ;

     for (int i = ; i <= n; ++i)
         sum += b[i];

     printf("%d\n", sum - maxFlow());
 }

@Author: YouSiki