寒假算法学习 H (8). 过河卒 （双向搜索优化DFS） B (2). 听歌识曲（前缀和+lower_bound）

H (8). 过河卒 （双向搜索优化DFS）

题目描述

---

棋盘上 A 点有一个过河卒，需要走到目标 B 点。卒行走的规则：可以向下、或者向右。同时在棋盘上 C 点有一个对方的马，该马所在的点和所有跳跃一步可达的点称为对方马的控制点。因此称之为“马拦过河卒”。

棋盘用坐标表示，A 点 (0,0)、B 点 (n,m)，同样马的位置坐标是需要给出的。

现在要求你计算出卒从 A 点能够到达 B 点的路径的条数，假设马的位置是固定不动的，并不是卒走一步马走一步。

洛谷：P1002 [NOIP2002 普及组] 过河卒 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

---

输入

---

一行四个正整数，分别表示 B 点坐标和马的坐标。

---

输出

---

一个整数，表示所有的路径条数。

---

样例输入

---

6 6 3 3

---

样例输出

---

6

---

题目分析

---

对于100% 的数据，1≤n,m≤20  0≤ 马的坐标，b≤20。

这个题的正解应该是线性DP，那好我们用搜索。

看到这个数据规模时就知道DFS肯定会TE，但是硬着头皮写了一下，最后不出所料TE了，之后就用学长刚教的双向搜索优化了一下（自己觉得并没有优化），轻松AC

---

60分代码

#include<bits/stdc++.h>
using namespace std;
#define fi first
#define se second
typedef pair<int,int> PII;
int cnt;
PII b;
PII h;
bool mp[21][21];
int dirh[10][2]= {{0,0},{1,2},{2,1},{-1,2},{2,-1},{-1,-2},{-2,-1},{-2,1},{1,-2}};
void init() {
	for(int i=0; i<9; i++) {
		int x=h.fi+dirh[i][0],y=h.se+dirh[i][1];
		if(x<=b.fi&&x>=0&&y<=b.fi&&y>=0)
			mp[x][y]=true;
	}
}//初始化，标记马的九个位置
bool judge(int x,int y) {
	return (x<=b.fi&&y<=b.se&&!mp[x][y]);
}//判断数据合法
void dfs(int x,int y) {
	if(x==b.fi&&y==b.se) {
		cnt++;
		return ;
	}//递归出口
	if(judge(x+1,y))
		dfs(x+1,y);//卒右移
	if(judge(x,y+1))
		dfs(x,y+1);//卒下移
}
int main() {
	cin>>b.fi>>b.se>>h.fi>>h.se;
	init();
	dfs(0,0);
	cout<<cnt;
	return 0;
}

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> PII;
#define fi first
#define se second

PII b,c;
ll f[21][21],n,cnt;
bool vis[21][21];
int dir[10][2]={0,0,1,2,2,1,-1,2,2,-1,-2,1,1,-2,-1,-2,-2,-1};

void init(){
	for(int i=0;i<9;i++){
		int x=c.fi+dir[i][0],y=c.se+dir[i][1];
		if(x>=0&&x<=b.fi&&y>=0&&y<=b.se)
			vis[x][y]=true;			
	}
}
bool judge(int x,int y){
	return (x>=0&&y>=0&&x<=b.fi&&y<=b.se&&!vis[x][y]);
}
void dfspre(int x,int y){
	if(x+y==n+1){
		f[x][y]++;
		return;
	}
	if(judge(x+1,y))
		dfspre(x+1,y);
	if(judge(x,y+1))
		dfspre(x,y+1);
}
void dfspost(int x,int y){
	if(x+y==n+1){
		cnt+=f[x][y];
		return;
	}
	if(judge(x-1,y))
		dfspost(x-1,y);
	if(judge(x,y-1))
		dfspost(x,y-1);
}
int main()
{
	cin>>b.fi>>b.se>>c.fi>>c.se;
	n=(b.fi+b.se)/2;
	init();
	dfspre(0,0);
	dfspost(b.fi,b.se);
	cout<<cnt;
	return 0;
}

 

(2). 听歌识曲（前缀和+lower_bound）

题目描述

---

洛洛有一份私人歌单，歌单里面塞满了他喜欢的歌曲，像夏恋、雨道、彩月、幻昼……整整有好几百首。洛洛每天都要把他的歌单听一遍，以致于他都能知道在什么时候放的是什么歌。
洛洛在向你推荐了他的歌单之后，决定考考你，从他的歌单开始播放起，第 t 秒正在播放的是第几首歌。

---

输入

---

第一行输入两个整数 n 和 t，分别表示歌单的歌曲总数以及第 t 秒播放哪首歌。
第二行有 n 个整数，A1， A2，...， An，分别表示歌单的第 i 首歌将会播放多长时间。

---

输出

---

输出一个整数，表示歌单按顺序播放后，第t秒播放的是第几首歌。

---

样例输入

---

3 5
5 5 5

---

样例输出

---

1

---

题目分析

---

一眼看出是前缀和，那么怎么确定t的位置呢，stl中的二分函数lower_bound()先生可以为我们解决问题，lower_bound查找第一个不下于t的地址，之后地址-f（前缀和数组的首地址）就能得到他的位置。

---

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+10;
ll f[N];
int n,t,a;
int main() {
	cin>>n>>t;
	for(int i=1; i<=n; i++) {
		cin>>a;
		f[i]=f[i-1]+a;
	}
	cout<<lower_bound(f,f+n,t)-f;
}