Budget 题目链接
题意:
有
n
∗
m
n*m
n∗m的数字矩阵,告诉你每行和每列的数字之和,接下来有
c
c
c行约束条件,第
c
c
c行会告诉你第
i
i
i行第
j
j
j列的格子应该有怎样的约束
(
>
,
<
,
=
)
(>,<,=)
(>,<,=),请输出任一种满足所有约束的矩阵或者输出不可能
思路:
经典的建图模型:
将每行每列都抽象为
1
1
1个点,源点与行相连,汇点与列相连,流量上下界均为
s
u
m
l
/
s
u
m
c
sum_l/sum_c
suml/sumc
每行每列直接的连线流量代表这些位置的约束范围
首先要保证所有的约束不会矛盾,否则无解。
建好图后在图上判断是否有可行流
最后根据图上每个位置流出的流量即可计算每个点可以的数字
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
#define inf 0x3f3f3f3f
#define double long double
using namespace std;
const int maxn=255;
int depth[maxn],now[maxn],num,head[maxn],s,t,n,m,ok,up[maxn][maxn],low[maxn][maxn],in[maxn],out[maxn];
int res[maxn][maxn];
struct road{int b,c,nex;}r[2000005];
void add(int a,int b,int c){r[num].b=b;r[num].c=c;r[num].nex=head[a];head[a]=num++;}
void addedge(int a,int b,char op,int val)
{
if(op=='=')
{
if(val<low[a][b] || val>up[a][b]) ok=0;
up[a][b]=low[a][b]=val;
}
else if(op=='>')
{
if(val>=up[a][b]) ok=0;
low[a][b]=max(low[a][b],val+1);
}
else
{
if(val<=low[a][b]) ok=0;
up[a][b]=min(up[a][b],val-1);
}
}
int make_level()
{
memset(depth,-1,sizeof(depth));
queue<int>q;
q.push(s);
depth[s]=1;
now[s]=head[s];
while(!q.empty())
{
int u=q.front();q.pop();
for(int i=head[u];~i;i=r[i].nex)
{
int v=r[i].b;
if(depth[v]!=-1 || r[i].c<=0) continue;
now[v]=head[v];
depth[v]=depth[u]+1;
q.push(v);
}
}
return depth[t]!=-1;
}
int dinic(int u,int flow)
{
if(u==t) return flow;
int sum=0;
for(int i=now[u];~i;i=r[i].nex)
{
now[u]=i;
int v=r[i].b;
if(depth[v]!=depth[u]+1 || r[i].c<=0) continue;
int use=dinic(v,min(flow-sum,r[i].c));
if(use)
{
r[i].c-=use;
r[i^1].c+=use;
sum+=use;
}
if(sum==flow) return flow;
}
if(sum==0) depth[u]=-1;
return sum;
}
int main()
{
int T;scanf("%d",&T);
while(T--)
{
memset(head,-1,sizeof(head));num=0;ok=1;
memset(low,0,sizeof(low));
memset(up,0x3f,sizeof(up));
memset(in,0,sizeof(in));
memset(out,0,sizeof(out));
scanf("%d%d",&n,&m);
s=m+n+1,t=s+1;
for(int i=1;i<=n;i++)
{
int x;scanf("%d",&x);
out[s]+=x;in[i]+=x;
}
for(int i=1;i<=m;i++)
{
int x;scanf("%d",&x);
out[i+n]+=x;in[t]+=x;
}
int k;scanf("%d",&k);
while(k--)
{
int a,b,v;char op[2];
scanf("%d%d%s%d",&a,&b,op,&v);
if(a&&b)
{
addedge(a,b,op[0],v);
}
else if(a)
{
for(int i=1;i<=m;i++) addedge(a,i,op[0],v);
}
else if(b)
{
for(int i=1;i<=n;i++) addedge(i,b,op[0],v);
}
else
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
addedge(i,j,op[0],v);
}
}
}
}
if(!ok)
{
puts("IMPOSSIBLE");continue;
}
int s2=t+1,t2=s2+1,sum=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
int ff=up[i][j]-low[i][j];
add(i,j+n,ff);add(j+n,i,0);
out[i]+=low[i][j];in[j+n]+=low[i][j];
}
}
for(int i=1;i<=n+m+2;i++)
{
if(out[i]>in[i]) sum+=(out[i]-in[i]),add(i,t2,(out[i]-in[i])),add(t2,i,0);
else if(in[i]>out[i]) add(s2,i,in[i]-out[i]),add(i,s2,0);
}
add(t,s,inf);add(s,t,0);
s=s2;t=t2;
int ans=0;
while(make_level()) ans+=dinic(s,inf);
if(ans!=sum)
{
puts("IMPOSSIBLE");continue;
}
for(int i=1;i<=n;i++)
{
for(int j=head[i];~j;j=r[j].nex)
{
int nx=r[j].b-n;
if(nx<1 || nx>m) continue;
res[i][nx]=low[i][nx]+r[1^j].c;
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
printf("%d ",res[i][j]);
}
puts("");
}
}
}