332.重新安排行程
本题难点在于容器的选择和使用上,既要找到一个对数据进行排序的容器,而且还要容易增删元素,迭代器还不能失效。一个机场映射多个机场用unordered_map,让多个机场之间有序,就是用map、multimap或者multiset,如果用multiset的话,遍历的时候不能删除元素,一旦删除元素的话,迭代器就失效了,所以用map。
class Solution {
private:
unordered_map<string,map<string,int>> targets;
bool backtracking(int ticketNum,vector<string>& res){
if(res.size()==ticketNum+1) return true;
for(pair<const string,int>& target:targets[res[res.size()-1]]){
if(target.second>0){
res.push_back(target.first);
target.second--;
if(backtracking(ticketNum,res)) return true;
target.second++;
res.pop_back();
}
}
//遍历了一遍都用过了
return false;
}
public:
vector<string> findItinerary(vector<vector<string>>& tickets) {
targets.clear();
vector<string> res;
for(const vector<string>& vec:tickets){
targets[vec[0]][vec[1]]++;//从vec[0]出发,到vec[1]的机票加1
}
//已知始发点是JFK,先放到结果集
res.push_back("JFK");
backtracking(tickets.size(),res);
return res;
}
};
51. N皇后
每一行没有进行判断,因为在单层搜索的过程中,每一层递归,只会选for循环(也就是同一行)里的一个元素,所以不用去重了。
class Solution {
public:
vector<vector<string>> res;
bool isValid(int row,int col,vector<string>& chessBoard,int n){
//看同一列
for(int i=0;i<row;i++){
if(chessBoard[i][col]=='Q') return false;
}
for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--){
if(chessBoard[i][j]=='Q') return false;
}
for(int i=row-1,j=col+1;i>=0&&j<n;i--,j++){
if(chessBoard[i][j]=='Q') return false;
}
return true;
}
void backtracking(vector<string>& chessBoard,int n,int row){
if(row==n){
res.push_back(chessBoard);
return;
}
for(int i=0;i<n;i++){//这每行就取一个元素,所以不用担心去重
if(isValid(row,i,chessBoard,n)){
chessBoard[row][i]='Q';
backtracking(chessBoard,n,row+1);
chessBoard[row][i]='.';
}
}
return;
}
vector<vector<string>> solveNQueens(int n) {
res.clear();
vector<string> chessBoard(n,string(n,'.'));
backtracking(chessBoard,n,0);
return res;
}
};
37. 解数独
注意是从数字1-9,不包含0.
class Solution {
public:
bool isValid(int row,int col,char ch,vector<vector<char>>& board){
for(int j=0;j<9;j++){
if(board[row][j]==ch) return false;
}
for(int i=0;i<9;i++){
if(board[i][col]==ch) return false;
}
int startrow=(row/3)*3;
int startcol=(col/3)*3;
for(int i=startrow;i<startrow+3;i++){
for(int j=startcol;j<startcol+3;j++){
if(board[i][j]==ch) return false;
}
}
return true;
}
bool backtracking(vector<vector<char>>& board){
for(int i=0;i<board.size();i++){
for(int j=0;j<board[0].size();j++){
if(board[i][j]=='.'){
for(char ch='1';ch<='9';ch++){//题目中说了是从数字1-9,没有0
if(isValid(i,j,ch,board)){
board[i][j]=ch;
bool res=backtracking(board);
if(res==true) return true;
board[i][j]='.';
}
}
//如果0到9都不行,就走不通
return false;
}
}
}
//如果没有空格了
return true;
}
void solveSudoku(vector<vector<char>>& board) {
backtracking(board);
}
};