题目描述
题解
很好想的最小割
将每个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求的是和的两倍,求一倍的话似乎更方便而且不容易出错。