题意,给一个矩阵,告诉你每行和、每列和,并且限制所填数不大于k,问矩阵是否唯一。
经典建图不说了,第一次遇到判断最大流唯一性的,学习了:用dfs来判断残网中是否还存在环,若存在,则表明绕这个环走一圈,(流一圈流量),还是最大流保持不变,说明还有解。输出方案就EASY了。
WA了一天:第一TLE,因为这题卡DINIC,我的没有优化,后来在zz1215学长加了一行代码,在增广的时候,若发现最小总流量已经为0,则标记该点层-1(不必要往下)。效果显著。原因2:判断环的时候,dfs判断环写错有木有!不可原谅a!每次枚举每个入口点(感觉可以优化),判断环,遇到子孩子是祖先就证明有环。
#include<cstdio> //600ms
#include<iostream>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
const int maxv=910;
const int maxe=405*405*2+450;
const int inf=0x3f3f3f3f;
int n,m,k;int allsumn=0,allsumm=0;
int nume=0;int e[maxe][4];int head[maxv];
int nsum[405];int msum[405]; bool flag;
void inline adde(int i,int j,int c)
{
e[nume][0]=j;e[nume][1]=head[i];head[i]=nume;
e[nume++][2]=c;
e[nume][0]=i;e[nume][1]=head[j];head[j]=nume;
e[nume++][2]=0;
}
int lev[maxv];int vis[maxv];
int ss=0;int tt=0;
bool bfs()
{
memset(lev,0,sizeof(lev));
memset(vis,0,sizeof(vis));
queue<int>q;
q.push(ss);
vis[ss]=1;
while(!q.empty())
{
int cur=q.front();
q.pop();
for(int i=head[cur];i!=-1;i=e[i][1])
{
int v=e[i][0];
if(e[i][2]>0&&!vis[v])
{
lev[v]=lev[cur]+1;
// if(v==tt)return 1; //这句不加,速度更快
q.push(v);
vis[v]=1;
}
}
}
return vis[tt];
}
int dfs(int u,int minf)
{
if(u==tt||minf==0)return minf;
int sumf=0,f;
for(int i=head[u];i!=-1&&minf;i=e[i][1])
{
int v=e[i][0];
if(lev[v]==lev[u]+1&&e[i][2]>0)
{
f=dfs(v,e[i][2]<minf?e[i][2]:minf);
sumf+=f;
e[i][2]-=f;e[i^1][2]+=f;
minf-=f;
}
}
if(!sumf)lev[u]=-1; //关键优化!
return sumf;
}
int dinic()
{
int sum=0;
while(bfs())
sum+=dfs(ss,inf);
return sum;
}
void init()
{
allsumm=allsumn=nume=0;
memset(head,-1,sizeof(head));
ss=n+m;tt=n+m+1;
}
void read_build()
{
for(int j=0;j<n;j++)
{
scanf("%d",&nsum[j]);
allsumn+=nsum[j];
}
for(int j=0;j<m;j++)
{
scanf("%d",&msum[j]);
allsumm+=msum[j];
}
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
adde(i,j+n,k);
} for(int i=0;i<n;i++)
{
adde(n+m,i,nsum[i]);
}
for(int i=0;i<m;i++)
{
adde(i+n,n+m+1,msum[i]);
} }
bool dfs_getother_ans(int u,int fa)
{
if(flag)return 1;
for(int i=head[u];i!=-1;i=e[i][1])
{
int v=e[i][0];
if(v==fa||e[i][2]<=0||v==n+m||v==n+m+1)continue;
if(!vis[v])
{
vis[v]=1;
dfs_getother_ans(v,u);
vis[v]=0; //出来的时候标记回来啊!
}
else
{
flag=1;
return 1;
}
if(flag)return 1;
}
return 0;
}
int ansjz[405][405];
int main()
{
while(scanf("%d%d%d",&n,&m,&k)!=EOF)
{
init();
read_build();
if(allsumm!=allsumn)
{printf("Impossible\n");continue;}
int ans=dinic();
if(ans!=allsumm)
{
printf("Impossible\n");
}
else
{
flag=0;
for(int i=0;i<n;i++)
{
vis[i]=1;
if(dfs_getother_ans(i,-1))
{
flag=1;
break;
}
vis[i]=0;
}
if(flag)
printf("Not Unique\n");
else
{
printf("Unique\n");
for(int i=0;i<=n-1;i++)
for(int j=head[i];j!=-1;j=e[j][1])
{
if(j%2==0)
{
ansjz[i][e[j][0]-n]=k-e[j][2];
}
}
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
if(j==m-1)printf("%d\n",ansjz[i][j]);
else printf("%d ",ansjz[i][j]);
}
}
} return 0;
}