poj 1698 SAP 模板刷题

#include<stdio.h> #include<memory.h> #include<iostream> using namespace std; const int MAXNUM = 2100000000 ; //权值上限 const int MAXPT = 400 ; //顶点数上限 const int MAXEG = MAXPT*MAXPT*2 ; //边数上限 const int MAXQUE = 200000 ; /* s = 1 ; // 源点 t = n ; // 汇点 */ class MNF_SAP { private: int s,t; int d[MAXPT]; //距离标号 int pre[MAXPT]; //前置顶点 int flow[MAXPT]; //到当前点为止，所有弧的最小值 int cur[MAXPT]; //当前弧 int cnt[MAXPT]; //k标号出现次数 int que[MAXPT*2],ft,rt; bool vis[MAXPT]; void init_d() { int i,u; memset(vis,0,sizeof(vis)); ft=rt=0; d[t]=0; vis[t]=true; que[++rt]=t; while(ft!=rt) { u=que[(++ft)%MAXQUE]; for(i=gra[u]; i; i=E[i].next) if(!vis[E[i].v] && !E[i].w) { d[E[i].v]=d[u]+1; vis[E[i].v]=true; que[(++rt)%MAXQUE]=E[i].v; } } } public: struct { int v,w,next; } E[MAXEG]; int n; int k; int gra[MAXPT]; MNF_SAP(){} ~MNF_SAP(){} void adde(int x,int y,int z) { E[++k].v = y ; E[k].next = gra[x] ; E[k].w = z ; gra[x] = k ; E[++k].v = x ; E[k].next = gra[y] ; E[k].w = 0 ; gra[y] = k ; } int SAP() { int u,v,i,maxflow=0; //总最大流 s=1; t=n; u=s; flow[s]=MAXNUM; for(i=1; i<=n; i++) cur[i]=gra[i]; //当前弧初始化 init_d(); for(i=1; i<=n; i++) cnt[d[i]]++; cnt[0]=n; while(d[s]<n) { for(i=cur[u]; i; i=E[i].next) //找允许弧 if(E[i].w && d[E[i].v]+1==d[u]) break; if(i) //存在允许弧 { cur[u]=i; //设置当前弧 v=E[i].v; if(E[i].w<flow[u]) flow[v]=E[i].w; else flow[v]=flow[u]; //标记当前顶点为止经过的最小弧 u=v; pre[v]=i; //前置顶点边号 if(u==t) { do { E[pre[u]].w-=flow[t]; //正向弧减a[t] E[pre[u]^1].w+=flow[t]; //通过异或操作找反向弧 u=E[pre[u]^1].v; } while(u!=s); maxflow+=flow[t]; //memset(flow,0,sizeof(flow)); flow[s]=MAXNUM; } } else //不存在允许弧 { if(--cnt[d[u]]==0) break; //间隙优化 d[u]=n; cur[u]=gra[u]; for(i=gra[u]; i; i=E[i].next) if(E[i].w && d[E[i].v]+1<d[u]) d[u]=d[E[i].v]+1; //修改距离标号为 最小的非允许弧加1 cnt[d[u]]++; if(u!=s) u=E[pre[u]^1].v; } } return maxflow; } } ; int temnode[51][8] ; int main() { MNF_SAP ob ; int nt ; int nodei ; int x,y,z; int fi; int i,j; int alldays ; int nf,days[55] ,ws; cin>>nt; while(nt--) { cin>>nf ; ob.k = 1 ; memset(ob.gra,0,sizeof(ob.gra)); memset(temnode,0,sizeof(temnode)); int wk[7] ; alldays = 0 ; nodei = nf +1 ; for(fi=1+1; fi<=nf+1; fi++) { for(i=0; i<7; i++) { scanf("%d",&wk[i]); } scanf("%d%d",&days[fi-2],&ws); alldays += days[fi-2] ; for(i=0; i<7; i++) { if(wk[i]) { for(j=0; j<ws; j++) { if( ! temnode[j][i] ) { temnode[j][i] = (++nodei) ; // the source -> which day x = 1; y = temnode[j][i]; z = 1 ; ob.adde(x,y,z); } x = temnode[j][i]; y = fi ; z = 1 ; // which day -> the film ob.adde(x,y,z); } } } } // the film -> sink for(fi=1+1; fi<=nf+1; fi++) { x = fi; y = nodei+1; z = days[fi-2]; ob.adde(x,y,z); } ob.n = nodei+1 ; int judge = ob.SAP(); if( alldays== judge ) { cout<<"Yes"<<endl; } else { cout<<"No"<<endl; } } return 0; }  

 

 

此题，一开始想构图，后来出现了错误在构图的时候，首先是自己构图的时候没有注意到有些点已经加入了，而我重新加入了，所以WA了

 

如下，简单的说一下构图，我构造了一个  源点s -> 哪一天 -> filmi -> 汇点t。

这里的权值，除了filmi -> 汇点t的权值是filmi需要的总天数，其他为1.