[BZOJ3275]Number（最小割）

题目描述

传送门

题解

很好想的最小割

将每个ai拆成xi，yi

s->xi,ai

yi->t,ai

对于不能同时选的两个点，xi->yi,INF

乱搞了一下很开心地A掉了，看hzwer的题解跟我差不多更开心了。。。

然而reflash说我做的不对？

网上有一种奇偶的方法？

会了补上。。。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#include<cmath>
using namespace std;

const int max_n=3005;
const int max_N=max_n*2+2;
const int max_m=max_n*1000;
const int max_e=max_m*2;
const int INF=1e9;

int n,a[max_n],N,sum,maxflow;
int tot,point[max_N],next[max_e],v[max_e],remain[max_e];
int deep[max_N],cur[max_N];
queue <int> q;

inline int in(){
int x=0; char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x;
}

inline int gcd(int a,int b){
if (!b) return a;
else return gcd(b,a%b);
}
inline bool check(int a,int b){
if (gcd(a,b)!=1) return false;
int c=a*a+b*b;
int t=sqrt(c);
if (t*t!=c) return false;
return true;
}

inline void addedge(int x,int y,int cap){
++tot; next[tot]=point[x]; point[x]=tot; v[tot]=y; remain[tot]=cap;
++tot; next[tot]=point[y]; point[y]=tot; v[tot]=x; remain[tot]=0;
}
inline bool bfs(int s,int t){
for (int i=1;i<=N;++i)
deep[i]=INF+1;
deep[s]=0;
for (int i=1;i<=N;++i)
cur[i]=point[i];
while (!q.empty()) q.pop();
q.push(s);

while (!q.empty()){
int now=q.front(); q.pop();
for (int i=point[now];i!=-1;i=next[i])
if (deep[v[i]]>INF&&remain[i]){
deep[v[i]]=deep[now]+1;
q.push(v[i]);
}
}

return deep[t]<INF;
}
inline int dfs(int now,int t,int limit){
if (now==t||!limit) return limit;
int flow=0,f;
for (int i=cur[now];i!=-1;i=next[i]){
cur[now]=i;
if (deep[v[i]]==deep[now]+1&&(f=dfs(v[i],t,min(remain[i],limit)))){
flow+=f;
limit-=f;
remain[i]-=f;
remain[i^1]+=f;
if (!limit) break;
}
}
return flow;
}
inline void dinic(int s,int t){
while (bfs(s,t))
maxflow+=dfs(s,t,INF);
}
int main(){
tot=-1;
memset(point,-1,sizeof(point));
memset(next,-1,sizeof(next));

n=in();
for (int i=1;i<=n;++i)
a[i]=in(),sum+=a[i];
N=n*2+2;
for (int i=1;i<=n;++i)
addedge(1,1+i,a[i]),addedge(1+n+i,N,a[i]);
for (int i=1;i<n;++i)
for (int j=i+1;j<=n;++j){
if (check(a[i],a[j])){
addedge(1+i,1+n+j,INF);
addedge(1+j,1+n+i,INF);
}
}
dinic(1,N);
printf("%d\n",sum-(maxflow>>1));
}

总结

刚开始好像炸了int

因为sum求的是和的两倍，求一倍的话似乎更方便而且不容易出错。