方方正正 |
Description |
一个r行c列的矩阵里的所有元素都为0或1，给出这个矩阵每一行的和以及每一列的和，那么是否存在这样一个矩阵满足条件呢，如果存在任意一个满足条件的矩阵则输出YES，如果不存在则输出NO？ |
Input |
每组测试数据第一行包含两个整数r,c,表示矩阵的行数和列数。 第二行包含r个32位无符号数，表示矩阵每行的和。 第三行包含c个32位无符号数，表示矩阵每列的和。 （1 <= r,c <= 100000) 处理到文件结束 |
Output |
如果存在这样的一个01矩阵，输出YES，否则输出NO。 |
Sample Input |
1 1 0 1 1 1 1 1 |
Sample Output |
NO YES |
Source |
2016级新生程序设计全国邀请赛 |
Time Limit: 1000 MS | Memory Limit: 32768 K |
题解：容易发现，
1，如果每行的和不等于每列的和，这是不可能的。
2，如果行的和的最大值大于列的非0的个数，这也是不可能的。
3，如果列的和的最大值大于行的非0的个数，这也是不可能的。
AC代码：
#include<cstdio>
#include<iostream>
#include<map>
#include<algorithm>
#include<cstring>
typedef long long ll;
using namespace std;
ll dp[110000];
int main()
{
int r,c;
while(~scanf("%d %d",&r,&c))
{
ll a=0,b=0,d=0;
ll a1=0,b1=0,d1=0;
ll tmp;
for(int i=1;i<=r;i++) //行
{
cin>>tmp;
if(tmp>0) a++;
b+=tmp;
if(tmp>d) d=tmp;
}
for(int i=1;i<=c;i++) //列
{
cin>>tmp;
if(tmp>0) a1++;
b1+=tmp;
if(tmp>d1) d1=tmp;
}
if( b!=b1 || a<d1 || a1<d )puts("NO");
else puts("YES");
}
return 0;
}
