美团codeM资格赛 送外卖

送外卖

时间限制：1秒

空间限制：32768K

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

输出描述:

输出一行字符串表示答案。

输入例子:

7
5 -3 6 5 -5 -1 6
-6 1 4 -2 0 -2 0

输出例子:

abbbb

解题思路：先构建反向图（本来是3能到6的，我加一条6到3的边），从n出发，广度优先搜索（bfs）一遍，能到1就有解。同时记录下哪些点被经过了，这在原图中，就是能到n的点。然后从1出发，每个点贪心选择：如果走a，这个点能到n，那就走a，否则走b。但如果走回到之前走过的点，那说明答案是Infinity。

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
#include <set>
#include <stack>
#include <map>
#include <climits>

using namespace std;

#define LL long long
const int INF = 0x3f3f3f3f;

vector<int> g[100005];
int a[100005],b[100005],n;
bool vis[100005];
bool vis1[100005];
char ch[100005];

void bfs()
{
    queue<int> q;
    vis[n]=1;
    q.push(n);
    while(!q.empty())
    {
        int pre=q.front();
        q.pop();
        int Size=g[pre].size();
        for(int i=0; i<Size; i++)
        {
            if(!vis[g[pre][i]])
            {
                vis[g[pre][i]]=1;
                q.push(g[pre][i]);
            }
        }
    }
}

int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++) g[i].clear();
        memset(vis,0,sizeof vis);
        memset(vis1,0,sizeof vis1);
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&a[i]);
            int k=i+a[i];
            if(k>=1&&k<=n) g[k].push_back(i);
        }
        for(int i=1; i<=n; i++)
        {
            scanf("%d",&b[i]);
            int k=i+b[i];
            if(k>=1&&k<=n) g[k].push_back(i);
        }
        bfs();
        if(!vis[1]) {printf("No solution!\n");continue;}
        int p=0;
        vis1[1]=1;
        bool flag=0;
        for(int x=1; x!=n&&!flag;)
        {
            int xx=x+a[x];
            if(xx>=1&&xx<=n&&vis[xx])
            {
                if(!vis1[xx])
                {
                    vis1[xx]=1;
                    ch[p++]='a';
                }
                else flag=1;
                x=xx;
            }
            else
            {
                xx=x+b[x];
                if(xx>=1&&xx<=n&&vis[xx])
                {
                    if(!vis1[xx])
                    {
                        vis1[xx]=1;
                        ch[p++]='b';
                    }
                    else flag=1;
                }
                else
                {
                    flag=2;
                    break;
                }
                x=xx;
            }
        }
        ch[p]='\0';
        if(!flag) printf("%s\n",ch);
        else if(flag==1) printf("Infinity!\n");
        else printf("No solution!\n");
    }
    return 0;
}