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;
}