小王在一条笔直的路上散步只能往前走和往后走,每走一段就会停下来查看自己所走的距离,查看之后小王可以调转方向,问小王在这条路上走可以由几个终点
输入:n表示 这条笔直路上的1-n个坐标点 m表示小王上传的距离数
输出:小王可能的终点数
思路:以每个点为起点进行搜索,查看小王走完所有距离时的终点,将其插入set之中自动去重,返回set集的大小就是终点的数目
#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
#include<set>
using namespace std;
void DFS(vector<int>& dis,set<int>& res,int index,int nums,int &n) {
if (nums == dis.size())
res.insert(index);
static const int dx[] = {1 ,-1 };//前后两个方向
for (int i = nums; i < dis.size(); ++i) {
for (int j = 0; j < 2; ++j) {
int newIndex = index + dx[j] * dis[i];
if (newIndex < 0 || newIndex >= n)//如果越界就换个方向
continue;
DFS(dis, res, newIndex, nums+1,n);
}
}
}
int main() {
int n, m;
cin >> n >> m;
int i = 0;
vector<int> dis(m,0);
while (i < m) {
cin >> dis[i++];
}
set<int> res;
for (i = 0; i < n; ++i) {
DFS( dis, res, i, 0,n);
}
cout << res.size() << endl;
return 0;
}