飞机票 dfs leetcode 332

题目

给定一个机票的字符串二维数组 [from, to]，子数组中的两个成员分别表示飞机出发和降落的机场地点，对该行程进行重新规划排序。所有这些机票都属于一个从 JFK（肯尼迪国际机场）出发的先生，所以该行程必须从 JFK 开始。
提示:
如果存在多种有效的行程，请你按字符自然排序返回最小的行程组合。例如，行程 [“JFK”, “LGA”] 与 [“JFK”, “LGB”] 相比就更小，排序更靠前
所有的机场都用三个大写字母表示（机场代码）。
假定所有机票至少存在一种合理的行程。
所有的机票必须都用一次 且 只能用一次。
示例 1：
输入：[[“MUC”, “LHR”], [“JFK”, “MUC”], [“SFO”, “SJC”], [“LHR”, “SFO”]]
输出：[“JFK”, “MUC”, “LHR”, “SFO”, “SJC”]
示例 2：
输入：[[“JFK”,“SFO”],[“JFK”,“ATL”],[“SFO”,“ATL”],[“ATL”,“JFK”],[“ATL”,“SFO”]]
输出：[“JFK”,“ATL”,“JFK”,“SFO”,“ATL”,“SFO”]
解释：另一种有效的行程是 [“JFK”,“SFO”,“ATL”,“JFK”,“ATL”,“SFO”]。但是它自然排序更大更靠后。

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        sort(tickets.begin(), tickets.end(), [](vector<string>& a, vector<string>& b) {
            return a[0] < b[0] || (a[0] == b[0] && a[1] < b[1]);
            });
        vector<string> ret(1, "JFK");
        const vector<string> temp;
        vector<bool> used(tickets.size(), false);//学习一下
        int flag =0;
        dfs(tickets, used, ret,flag);
        return ret;
    }
private: 
    void  dfs(vector<vector<string>>& tickets, vector<bool>& used, vector<string>& ret,int& flag) {
        if (ret.size() >= tickets.size() + 1) {flag=1; return ;}
        string c = ret.back();
        for(int i=0;i<tickets.size();i++)
        {
            if(!used[i]&& tickets[i][0]==c)
            {
                used[i] = true;
                ret.push_back(tickets[i][1]);
                dfs(tickets,used,ret,flag);
                if(flag ==1) return;
                used[i] = false;
                ret.pop_back();
            }
        }
    }
};

*主要的难点在于当得到答案后不知道如何保存答案不被修改，我刚开始借鉴的那段代码 dfs函数是bool类型，精妙但不符合我的思维，所以我新加上了flag，来保护最后结果。
dfs类型的题目，有一些是压入answer栈，之后的操作不影响answer内已存在的值，answer大多是二维的，但有一些是一维的答案，会受影响，可能最好的做法是通过bool dfs解决问题吧。但我不想去学。。
现将bool dfs置于下方。

借鉴代码：

bool dfs(vector<vector<string>>& tickets, vector<bool>& used, vector<string>& ret) {
        if (ret.size() >= tickets.size() + 1) return true;
        string c = ret.back();
        for (int i = 0; i < tickets.size(); i++) {
            if (tickets[i][0] != c || used[i]) continue;
            used[i] = true;
            ret.push_back(tickets[i][1]);
            if (dfs(tickets, used, ret)) return true;
            ret.pop_back();
            used[i] = false;
        }
        return false;
    }

其中：if (dfs(tickets, used, ret)) return true;起到了保护答案的作用。