送外卖 牛客网 DFS

链接：https://ac.nowcoder.com/acm/problem/13224
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld
题目描述
n 个小区排成一列，编号为从 0 到 n-1 。一开始，美团外卖员在第0号小区，目标为位于第 n-1 个小区的配送站。
给定两个整数数列 a[0]~a[n-1] 和 b[0]~b[n-1] ，在每个小区 i 里你有两种选择：

1. 选择a：向前 a[i] 个小区。
2. 选择b：向前 b[i] 个小区。

把每步的选择写成一个关于字符 ‘a’ 和 ‘b’ 的字符串。求到达小区n-1的方案中，字典序最小的字符串。如果做出某个选择时，你跳出了这n个小区的范围，则这个选择不合法。
• 当没有合法的选择序列时，输出 “No solution!”。
• 当字典序最小的字符串无限长时，输出 “Infinity!”。
• 否则，输出这个选择字符串。

字典序定义如下：串s和串t，如果串 s 字典序比串 t 小，则
• 存在整数 i ≥ -1，使得∀j，0 ≤ j ≤ i，满足s[j] = t[j] 且 s[i+1] < t[i+1]。
• 其中，空字符 < ‘a’ < ‘b’。
输入描述:
输入有 3 行。

第一行输入一个整数 n (1 ≤ n ≤ 10^5)。

第二行输入 n 个整数，分别表示 a[i] 。

第三行输入 n 个整数，分别表示 b[i] 。

−n ≤ a[i], b[i] ≤ n
输出描述:
输出一行字符串表示答案。
示例1
输入
复制
7
5 -3 6 5 -5 -1 6
-6 1 4 -2 0 -2 0
输出
复制
abbbb

思路：从初始位置开始遍历，无非就是选a或b，题目要求选择字典序最小，所以dfs的时候先a后b，然后又因为分为三种情况：
1：No solution!
2：存在字典序最小的步骤
3：Infinity!
所以定义一个vis数组
vis[pos] == 2 表示属于刚刚走过，正在尝试经过该位置看能不能字典序最小的解，或者Infinity!
vis[pos] == 1 表示属于走过这个路程，并且找不到字典序最小的解，或者Infinity!

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<ctime>
using namespace std;

typedef long long ll;

const int maxn = 4 * 1e7 + 100;
const int mod = 1e9 + 7;

ll n,flag,a[maxn],b[maxn];
int vis[maxn];
int huan[maxn];
char ans[maxn];


bool dfs(ll pos, ll cur)
{
    if(pos < 1 || pos > n) return false;
    if(pos == n){
        ans[cur] = '\0'; flag = 1; return true;
    }
    if(vis[pos] == 1) return false;
    else if(vis[pos] == 2){ huan[pos] = 1; return false;}
    vis[pos] += 2;
    ans[cur] = 'a';
    if(dfs(pos+a[pos], cur+1)){
        if(huan[pos]) flag = 2;
        return true;
    }
    ans[cur] = 'b';
    if(dfs(pos+b[pos], cur+1)){
        if(huan[pos]) flag = 2;
        return true;
    }
    vis[pos] --;
    return false;
}

int main()
{
    //freopen("input.c", "r", stdin);
    cin>>n;
    for(int i = 1; i <= n; i ++) cin>>a[i];
    for(int i = 1; i <= n; i ++) cin>>b[i];
    dfs(1,0);
    if(flag == 0) cout<<"No solution!"<<endl;
    else if(flag == 1) cout<<ans<<endl;
    else if(flag == 2) cout<<"Infinity!"<<endl;
    return 0;
}