主要是参考别人的代码和想法做的
这题是网络流建模的题
奇偶性建模,主要是用二分图的形式,奇数二分图为一边,偶数为另外一边,主要是要注意1。 我将我的dinic模板抽象为下面这样,仅供参考。
主要是用前向星的方式存边 struct E{ int to,next,w; }edge[maxn*200]; int head[maxn]; void init() { memset(head,-1,sizeof(head)); } void addedge(int u,int v,int w) { edge[++tot].to=v; edge[tot].w=w; edge[tot].next=head[u]; head[u]=tot; edge[++tot].to=u; edge[tot].w=0; edge[tot].next=head[v]; head[v]=tot; } void build_graph() { init();//初始化 //这一部分主要是用来建图 } int level[maxn],s,t; bool bfs() { queue<int>q; memset(level,-1,sizeof(level)); level[s]=0; q.push(s); while(!q.empty()){ int u=q.front();q.pop(); for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(level[v]==-1&&edge[i].w>0){ level[v]=level[u]+1; q.push(v); } } } if(level[t]==-1) return false; else return true; } int dfs(int u,int flow) { if(u==t) return flow; int tmp, res = 0; for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(level[v]==level[u]+1&&edge[i].w>0){ tmp=dfs(v,min(edge[i].w,flow)); edge[i].w-=tmp; edge[i^1].w+=tmp; res += tmp; flow -= tmp; } } return res; } int dinic() { int flow=0; while (bfs()) { flow += dfs(s,inf); } return flow; }下面是这题的代码
#include <bits/stdc++.h> using namespace std; typedef long long LL; const LL INF=1e17L; const int inf=0x3f3f3f3f; const int maxn=1e2+5; const int mod=1e9+7; int n,k,cnt=0; struct node{ int p,c,l; }a[maxn]; bool isprime(int x) { if(x==2) return true; int t=sqrt(x); for(int i=2;i<=t;i++) if(x%i==0) return false; return true; } struct E{ int to,next,w; }edge[maxn*200]; int head[maxn],tot,s,t,total; void init() { memset(head,-1,sizeof(head)); tot=-1; } void addedge(int u,int v,int w) { edge[++tot].to=v; edge[tot].w=w; edge[tot].next=head[u]; head[u]=tot; edge[++tot].to=u; edge[tot].w=0; edge[tot].next=head[v]; head[v]=tot; } void build_graph(int mid) { s=0;t=n+1;total=0; init(); int p1=-1; for(int i=1;i<=n;i++){ if(a[i].c==1 && a[i].l<=mid){ if(p1==-1 || a[i].p>a[p1].p) p1=i; } } if(p1!=-1){ addedge(s,p1,a[p1].p); total+=a[p1].p; for(int i=1;i<=n;i++) if(a[i].c!=1 && isprime(1+a[i].c) && a[i].l<=mid) addedge(p1,i,inf); } for(int i=1;i<=n;i++){ if(a[i].c==1 || a[i].l>mid) continue; if(a[i].c&1) addedge(s,i,a[i].p); else addedge(i,t,a[i].p); total+=a[i].p; for(int j=i+1;j<=n;j++){ if(a[j].c==1 || a[j].l>mid || !isprime(a[i].c+a[j].c)) continue; if(a[i].c&1) addedge(i,j,inf); else addedge(j,i,inf); } } } int level[maxn]; bool bfs() { queue<int>q; memset(level,-1,sizeof(level)); level[s]=0; q.push(s); while(!q.empty()){ int u=q.front();q.pop(); for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(level[v]==-1&&edge[i].w>0){ level[v]=level[u]+1; q.push(v); } } } if(level[t]==-1) return false; else return true; } int dfs(int u,int flow) { if(u==t) return flow; int tmp, res = 0; for(int i=head[u];i!=-1;i=edge[i].next){ int v=edge[i].to; if(level[v]==level[u]+1&&edge[i].w>0){ tmp=dfs(v,min(edge[i].w,flow)); edge[i].w-=tmp; edge[i^1].w+=tmp; res += tmp; flow -= tmp; } } return res; } int dinic() { int flow=0; while (bfs()) { flow += dfs(s,inf); } return flow; } int main() { cin>>n>>k; for(int i=1;i<=n;i++) cin>>a[i].p>>a[i].c>>a[i].l; int l=1,r=100,ans=-1; while(l<=r){ int mid=(l+r)>>1; build_graph(mid); int cut=dinic(); if(total-cut>=k){ ans=mid; r=mid-1; } else l=mid+1; } cout<<ans<<endl; return 0; }