将星期拆点,符合条件的连边,最后统计汇点流量是否满即可了,注意结点编号。
#include<cstdio> #include<cstring> #include<cmath> #include<iostream> #include<algorithm> #include<set> #include<map> #include<queue> #include<vector> #include<string> #define eps 1e-12 #define INF 0x7fffffff #define maxn 1000 using namespace std; int n,m; int en; int st,ed; //源点和汇点 int dis[maxn] ;//dis[i],表示 到 原点 s 的 层数 int que[999999]; int can[55][11]; struct edge { int to,c,next; }; edge e[999999]; int head[maxn]; void add(int a,int b,int c) { e[en].to=b; e[en].c=c; e[en].next=head[a]; head[a]=en++; e[en].to=a; e[en].c=0; e[en].next=head[b]; head[b]=en++; } int bfs() { memset(dis,-1,sizeof(dis)); dis[st]=0; int front=0,rear=0; que[rear++]=st; while(front<rear) { int j=que[front++]; for(int k=head[j];k!=-1;k=e[k].next) { int i=e[k].to; if(dis[i]==-1&&e[k].c) { dis[i] = dis[j]+ 1 ; que[rear++]=i; if(i==ed) return true; } } } return false; } int dfs(int x,int mx) { int i,a; if(x==ed) return mx ; int ret=0; for(int k=head[x];k!=-1&&ret<mx;k=e[k].next) { if(e[k].c&&dis[e[k].to]==dis[x]+1) { int dd=dfs(e[k].to,min(e[k].c,mx-ret)); e[k].c-=dd; e[k^1].c+=dd; ret+=dd; } } if(!ret) dis[x]=-1; return ret; } void init() { en=0; st=0; //源 ed=20+50*7+100; //汇 memset(head,-1,sizeof(head)); } void build() { for(int i=1;i<=n;i++) add(i,ed,can[i][8]); for(int i=n+1;i<=n+350;i++) add(st,i,1); for(int i=1;i<=n;i++) { for(int j=0,id=n+1;j<can[i][9];j++) //第几周 { for(int t=1;t<=7;t++,id++) { if(can[i][t]) add(id,i,1); } } } } int dinic() { int tmp=0; int maxflow=0; while(bfs()) { while(tmp=dfs(st,INF)) maxflow+=tmp; } return maxflow; } int main() { int cas; int ans; scanf("%d",&cas); while(cas--) { ans=0; scanf("%d",&n); for(int i=1;i<=n;i++) { for(int j=1;j<=9;j++) { scanf("%d",&can[i][j]); } ans+=can[i][8]; } init(); build(); // cout<<dinic()<<endl; printf("%s ",dinic()==ans?"Yes":"No"); } }