【BZOJ 4500 矩阵】
Time Limit: 1 Sec Memory Limit: 256 MB
Submit: 390 Solved: 217
[Submit][Status][Discuss]
Description
有一个n*m的矩阵,初始每个格子的权值都为0,可以对矩阵执行两种操作:
1. 选择一行, 该行每个格子的权值加1或减1。
2. 选择一列, 该列每个格子的权值加1或减1。
现在有K个限制,每个限制为一个三元组(x,y,c),代表格子(x,y)权值等于c。问是否存在一个操作序列,使得操作完后的矩阵满足所有的限制。如果存在输出”Yes”,否则输出”No”。
Input
先输入一个T(T <= 5)代表输入有T组数据,每组数据格式为:
第一行三个整数n, m, k (1 <= n, m,k <= 1000)。
接下来k行,每行三个整数x, y, c。
Output
对于每组数据,输出Yes或者No。
Sample Input
2
2 2 4
1 1 0
1 2 0
2 1 2
2 2 2
2 2 4
1 1 0
1 2 0
2 1 2
2 2 1
Sample Output
Yes
No
HINT
Source
【题解】
①行列差分约束。
②每个行每个列看做一个点。
③建立超级源点保证图连通,进行一次SPFA即可解决。
#include<queue> #include<stdio.h> #define inf 1000000007 #define go(i,a,b) for(int i=a;i<=b;i++) #define fo(i,a,x) for(int i=a[x],v=e[i].v;i;i=e[i].next,v=e[i].v) const int N=2010; struct E{int v,next,w;}e[N<<2]; int T,n,m,K,head[N],k,x,y,v,S,d[N],vis[N];bool inq[N],bad; void ADD(int u,int v,int w){e[k]=(E){v,head[u],w};head[u]=k++;} bool SPFA() { std::queue<int>q;d[S]=0; while(!q.empty())q.pop();q.push(S);int u; while(!q.empty()) { inq[u=q.front()]=0;q.pop(); fo(i,head,u)if(d[u]+e[i].w<d[v]) { d[v]=d[u]+e[i].w; if((++vis[v])>(n+m))return 0; if(!inq[v])inq[v]=1,q.push(v); } } return 1; } int main() { scanf("%d",&T); while(T--&&scanf("%d%d%d",&n,&m,&K)) { go(i,0,n+m)d[i]=inf,head[i]=inq[i]=vis[i]=0;bad=0;k=1; go(i,1,K)scanf("%d%d%d",&x,&y,&v),ADD(x,y+n,v),ADD(y+n,x,-v); go(i,1,n+m)ADD(S,i,0);puts(SPFA()?"Yes":"No"); } return 0; }//Paul_Guderian
.