在一个遥远的国度,一侧是风景秀美的湖泊,另一侧则是漫无边际的沙漠。该国的行政区划十分特殊,刚好构成一个
NN
N 行
×M \times M
×M 列的矩形,如上图所示,其中每个格子都代表一座城市,每座城市都有一个海拔高度。
为了使居民们都尽可能饮用到清澈的湖水,现在要在某些城市建造水利设施。水利设施有两种,分别为蓄水厂和输水站。蓄水厂的功能是利用水泵将湖泊中的水抽取到所在城市的蓄水池中。
因此,只有与湖泊毗邻的第
11
1 行的城市可以建造蓄水厂。而输水站的功能则是通过输水管线利用高度落差,将湖水从高处向低处输送。故一座城市能建造输水站的前提,是存在比它海拔更高且拥有公共边的相邻城市,已经建有水利设施。由于第
NN
N 行的城市靠近沙漠,是该国的干旱区,所以要求其中的每座城市都建有水利设施。那么,这个要求能否满足呢?如果能,请计算最少建造几个蓄水厂;如果不能,求干旱区中不可能建有水利设施的城市数目。
输入输出格式
输入格式:
每行两个数,之间用一个空格隔开。输入的第一行是两个正整数
N,MN,M
N,M ,表示矩形的规模。接下来
NN
N 行,每行
MM
M 个正整数,依次代表每座城市的海拔高度。
输出格式:
两行。如果能满足要求,输出的第一行是整数
11
1 ,第二行是一个整数,代表最少建造几个蓄水厂;如果不能满足要求,输出的第一行是整数
00
0 ,第二行是一个整数,代表有几座干旱区中的城市不可能建有水利设施。
输入输出样例
输入样例#1: 输出样例#1:
2 5 1
9 1 5 4 3 1
8 7 6 1 2
输入样例#2: 输出样例#2:
3 6 1
8 4 5 6 4 4 3
7 3 4 3 3 3
3 2 2 1 1 2
说明
【样例1 说明】
只需要在海拔为 9
的那座城市中建造蓄水厂,即可满足要求。
【样例2 说明】
上图中,在
3个粗线框出的城市中建造蓄水厂,可以满足要求。以这
3个蓄水厂为源头在干旱区中建造的输水站分别用3 种颜色标出。当然,建造方法可能不唯一。
这题一看就想到的是搜索
从第一行开始搜索将最后一行能搜索到的标记一下,就能判断最后一行能否有水利设施
#include <iostream>
#include <cstring>
using namespace std;
int n,m;
int a[105][105];
int b[105][105];
void dfs(int x,int y){
if(x<1||x>n||y<1||y>m){
return;
}
if(x==n){
b[n][y]=1;
}
if(a[x+1][y]<a[x][y]) dfs(x+1,y);
if(a[x-1][y]<a[x][y]) dfs(x-1,y);
if(a[x][y+1]<a[x][y]) dfs(x,y+1);
if(a[x][y-1]<a[x][y]) dfs(x,y-1);
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
}
memset(b,0,sizeof(b));
for(int i=1;i<=m;i++)
{
dfs(1,i);
}
for(int i=1;i<=m;i++)
{
if(b[n][i]!=1){
cout<<"0"<<endl;
return 0;
}
}
cout<<"1"<<endl;
return 0;
}
然后还要实现两个功能 1:如果能,计算最少建造几个蓄水厂2:如果不能,求干旱区中不可能建有水利设施的城市数目。
明显第二个要求容易实现,只需在之前的代码上进行一点修改,数出b[n][i]为0的个数即可
#include <iostream>
#include <cstring>
using namespace std;
int n,m;
int a[105][105];
int b[105][105];
void dfs(int x,int y){
if(x<1||x>n||y<1||y>m){
return;
}
if(x==n){
b[n][y]=1;
}
if(a[x+1][y]<a[x][y]) dfs(x+1,y);
if(a[x-1][y]<a[x][y]) dfs(x-1,y);
if(a[x][y+1]<a[x][y]) dfs(x,y+1);
if(a[x][y-1]<a[x][y]) dfs(x,y-1);
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
}
memset(b,0,sizeof(b));
for(int i=1;i<=m;i++)
{
dfs(1,i);
}
int num=0;
for(int i=1;i<=m;i++)
{
if(b[n][i]!=1){
num++;
}
}
if(num!=0){
cout<<"0"<<endl;
cout<<num<<endl;
return 0;
}else{
cout<<"1"<<endl;
}
return 0;
}
然后第一个功能的实现 我们可以将最后一行想象为一条大线段,开头一行下来覆盖的每一点连起来为一条小线段
那么问题就转化为线段覆盖问题
那么要对dfs增加一点操作记录每个能到达最底端的范围的左端和右端
void dfs(int x,int y,int z){
if(x<1||x>n||y<1||y>m){
for(int i=1;i<=m;i++){
if(b[n][i]==z){
if(i<c[o].left) c[o].left=i;
if(i>c[o].right) c[o].right=i;
}
}
return;
}
if(x==n){
b[n][y]=z;
}
if(a[x+1][y]<a[x][y]) dfs(x+1,y,z);
if(a[x-1][y]<a[x][y]) dfs(x-1,y,z);
if(a[x][y+1]<a[x][y]) dfs(x,y+1,z);
if(a[x][y-1]<a[x][y]) dfs(x,y-1,z);
然后把数据根据左端的大小从小到大排
然后依次寻找能够覆盖目前右端点的最大右端点
int fugai(){
sort(c+1,c+1+o,cmp);
int t=1;
int ans=0,i=1;
while(t<m)
{
ans++;
int s=t;
for(int i=1;i<=o;i++){
if(c[i].left<=s&&c[i].right>t){
t=c[i].right;
}
}
}
return ans;
}
把代码整合一下
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int n,m;
int a[505][505];
int b[505][505];
struct node{
int left;
int right;
}c[1005];
int o=1;
void dfs(int x,int y,int z){
if(x<1||x>n||y<1||y>m){
for(int i=1;i<=m;i++){
if(b[n][i]==z){
if(i<c[o].left) c[o].left=i;
if(i>c[o].right) c[o].right=i;
}
}
return;
}
if(x==n){
b[n][y]=z;
}
if(a[x+1][y]<a[x][y]) dfs(x+1,y,z);
if(a[x-1][y]<a[x][y]) dfs(x-1,y,z);
if(a[x][y+1]<a[x][y]) dfs(x,y+1,z);
if(a[x][y-1]<a[x][y]) dfs(x,y-1,z);
}
bool cmp(node x,node y){
return x.left<y.left;
}
int fugai(){
sort(c+1,c+1+o,cmp);
int t=1;
int ans=0,i=1;
while(t<m)
{
ans++;
int s=t;
for(int i=1;i<=o;i++){
if(c[i].left<=s&&c[i].right>t){
t=c[i].right;
}
}
}
return ans;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
}
memset(b,0,sizeof(b));
for(int i=1;i<=m;i++)
{
c[i].left=m;
c[i].right=1;
}
for(int i=1;i<=m;i++)
{
dfs(1,i,i);
if(c[o].left>c[o].right){
c[o].left=0;
c[o].right=0;
}
o++;
}
int num=0;
for(int i=1;i<=m;i++)
{
if(b[n][i]==0){
num++;
}
}
if(num!=0){
cout<<"0"<<endl;
cout<<num<<endl;
return 0;
}else{
cout<<"1"<<endl;
cout<<fugai()<<endl;
}
return 0;
}
这个代码在洛谷上只能过一个点,其余超时qwq,所以要想办法优化以下
仔细观察可以发现在开始那行开始的点两旁不能有比它大的点,不然就不需要进行搜索
所以进行优化
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int n,m;
int a[505][505];
int b[505][505];
struct node{
int left;
int right;
}c[1005];
int o=1;
void dfs(int x,int y,int z){
if(x<1||x>n||y<1||y>m){
for(int i=1;i<=m;i++){
if(b[n][i]==z){
if(i<c[o].left) c[o].left=i;
if(i>c[o].right) c[o].right=i;
}
}
return;
}
if(x==n)
{
b[n][y]=z;
}
if(a[x+1][y]<a[x][y]) dfs(x+1,y,z);
if(a[x-1][y]<a[x][y]) dfs(x-1,y,z);
if(a[x][y+1]<a[x][y]) dfs(x,y+1,z);
if(a[x][y-1]<a[x][y]) dfs(x,y-1,z);
}
bool cmp(node x,node y){
return x.left<y.left;
}
int fugai(){
sort(c+1,c+1+o,cmp);
int t=1;
int ans=0,i=1;
while(t<m)
{
ans++;
int s=t;
for(int i=1;i<=o;i++){
if(c[i].left<=s&&c[i].right>t){
t=c[i].right;
}
}
}
return ans;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>a[i][j];
}
memset(b,0,sizeof(b));
for(int i=1;i<=m;i++)
{
c[i].left=m;
c[i].right=1;
}
for(int i=1;i<=m;i++)
{
if(a[1][i]>=a[1][i-1]&&a[1][i]>=a[1][i+1])
{
dfs(1,i,i);
}
if(c[o].left>c[o].right){
c[o].left=0;
c[o].right=0;
}
o++;
}
for(int i=1;i<=m;i++)
{
cout<<b[n][i];
}
cout<<endl;
int num=0;
for(int i=1;i<=m;i++)
{
if(b[n][i]==0){
num++;
}
}
if(num!=0){
cout<<"0"<<endl;
cout<<num<<endl;
return 0;
}else{
cout<<"1"<<endl;
cout<<fugai()<<endl;
}
return 0;
}
qwq,虽然快了,但还是只能过一个点,qwq;
求大佬指教