51nod2993 Budget

2993 Budget

求一个的矩阵a，a中的元素都要求是非负数，关于矩阵中的元素大小有条限制，限制有两种类型：

"1 i j x" 表示

"2 i j x" 表示

还要求了每一行每一列的元素和的值，如果存在这样的矩阵则输出"Yes"，否则输出"No"

多组数据

输入

第一行一个整数case，表示数据组数
接下来case组数据，每组数据的第一行输入三个整数n,m,k表示矩阵的大小和限制个数
接下来一行n个数，第i个数表示第i行的数字和
接下来一行m个数，第j个数表示第j行的数字和
接下来k行，每行四个整数形如“1 i j x”或"2 i j x"含义如题目所述

输出

每组数据输出一行一个字符串“Yes”或“No”，分别表示存在这样的矩阵或者不存在。

数据范围

对于30%的数据: case=1,n,m<=20
对于60%的数据: case<=5,n,m<=20
对于100%的数据: case<=5,n<=200,m<=20,k<=1000,1<=x<=2000

输入样例

输入样例1:
1
8 11 6
5352 9717 15549 21582 27797 33935 38353 43483 
2929 6660 10775 16382 19525 24512 26925 30930 35155 39386 43483 
1 7 9 263
1 2 9 145
1 8 7 206
1 4 9 87
2 4 9 891
1 3 8 318
输入样例2:
1
9 18 1
7939 2225 5611 266 2866 5896 10975 269 12937 
4205 871 6155 3441 7485 6697 6784 856 2865 8001 3289 8401 7593 3805 6209 8485 2242 291 
2 9 1 646
输入样例3:
1
10 18 6
1671 15573 321 5781 8242 877 13151 2999 11098 9704 
6841 9841 2931 4593 7481 1956 5493 9505 881 9953 8349 6472 4376 6376 356 5431 9217 5689 
1 7 7 391
2 6 12 13
2 7 3 679
1 8 3 824
1 5 7 836
1 7 7 941

输出样例

输出样例1:
No
输出样例2:
Yes
输出样例3:
Yes

解析：有上下界可行流

放代码：

#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;
struct edge
{
	int v,w,nxt;
}e[10005];
struct node
{
	int u,v,minw,maxw;
}E[10005];
int h[305],dis[305],cur[305],in[305],out[305],a[205][25],b[205][25];
int T,tot,op,n,m,s,t,u,v,w,ans,cnt,sum,k,ss,tt;
void add(int u,int v,int w)
{
	e[cnt].v=v;
	e[cnt].w=w;
	e[cnt].nxt=h[u];
	h[u]=cnt++;
}
bool bfs()
{
	memset(dis,-1,sizeof(dis));
	dis[s]=0;
	queue<int>q;
	q.push(s);
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=h[u];i!=-1;i=e[i].nxt)
		{
			int v=e[i].v;
			if(dis[v]==-1&&e[i].w)
			{
				dis[v]=dis[u]+1;
				q.push(v);
			}
		}
	}
	return dis[t]!=-1;
}
int dfs(int u,int flow)
{
	if(u==t)return flow;
	int rest=flow;
	for(int i=cur[u];i!=-1;i=e[i].nxt)
	{
		cur[u]=i;
		int v=e[i].v;
		if(dis[v]==dis[u]+1&&e[i].w)
		{
			int d=dfs(v,min(rest,e[i].w));
			e[i].w-=d;e[i^1].w+=d;
			rest-=d;
			if(!rest)break;
		}
	}
	return flow-rest;
}
signed main()
{
    scanf("%d",&T);
    loop:
    while(T--)
    {
        scanf("%d%d%d",&n,&m,&k);
    	memset(h,-1,sizeof(h));
    	memset(in,0,sizeof(in));
    	memset(out,0,sizeof(out));
    	memset(a,inf,sizeof(a));
    	memset(b,0,sizeof(b));
    	s=n+m+1;t=n+m+2;
    	tot=0;ans=0;cnt=0;sum=0;
    	for(int i=1;i<=n;i++)
    	{
    	    scanf("%d",&w);
    	    E[++tot].u=s;
    	    E[tot].v=i;
    	    E[tot].minw=w;
    	    E[tot].maxw=w;
    	}
    	for(int i=1;i<=m;i++)
    	{
    	    scanf("%d",&w);
    	    E[++tot].u=i+n;
    	    E[tot].v=t;
    	    E[tot].minw=w;
    	    E[tot].maxw=w;
    	}
    	for(int i=1;i<=k;i++)
    	{
    	    scanf("%d%d%d%d",&op,&u,&v,&w);
    	    if(op==1)a[u][v]=min(a[u][v],w);
    	    else b[u][v]=max(b[u][v],w);
    	}
    	for(int i=1;i<=n;i++)
    	{
    	    for(int j=1;j<=m;j++)
    	    {
    	        E[++tot].u=i;
    	        E[tot].v=j+n;
    	        E[tot].minw=b[i][j];
    	        E[tot].maxw=a[i][j];
    	        if(b[i][j]>a[i][j])
    	        {
    	        	printf("No\n");
    	        	goto loop;
    	        }
    	    }
    	}
    	for(int i=1;i<=tot;i++)
    	{
    		in[E[i].v]+=E[i].minw;
    		out[E[i].u]+=E[i].minw;
    		add(E[i].u,E[i].v,E[i].maxw-E[i].minw);
    		add(E[i].v,E[i].u,0);
    	}
    	for(int i=1;i<=n+m;i++)
    	{
    		if(in[i]>=out[i])
    		{
    			add(s,i,in[i]-out[i]);
    			add(i,s,0);
    			sum+=in[i]-out[i];
    		}
    		else 
    		{
    			add(i,t,out[i]-in[i]);
    			add(t,i,0);
    		}
    	}
    	while(bfs())
    	{
    		for(int i=1;i<=n+m+2;i++)cur[i]=h[i];
    		ans+=dfs(s,inf);
    	}
    	if(ans!=sum)printf("No\n");
    	else printf("Yes\n");
    }
	return 0;
}