原本以為是一道很簡單的模擬題,結果寫了一個小時。。。很長時間不碰算法題,的確手感差很多。不過我覺得隨著刷題慢慢多起來應該會好的。
題目的意思也有點含糊,需要自己去猜,大概意思就是槽里有一堆木頭,每個槽剛開始的時候只有一個,需要移過來移過去,有四種方式,這四種方式都是針對木頭而言的,因此我們必須時刻記錄每個木頭的位置。當然還需要數據結構記錄槽的狀態,最后需要輸出。
四種移動方式有一些是共通的,因此需要將其抽象成函數,我這里抽象了三個函數:
void ret_block(int x);
將木頭x頭頂的木頭歸還到原本的槽里面(i
號木頭到i
號槽)
void mve(int x, int idx);
將木頭x
移動到槽idx
的位置,之所以不叫move
是因為不想和標準庫的move
函數沖突
void pile(int x, int idx);
將木頭x
以及其頭頂的所有木頭移動到槽idx
雖然使用三個函數簡化了四種操作,但是我覺得自己分離的不夠清晰,按道理講pile
函數應該調用mve
函數,因為一個是移動一個木頭,一個是移動一堆木頭,可是因為使用的是vector
,導致無法隨機插入。
順帶吐槽一下:我為了控制不輸出最后的換行專門寫了一個Newline
函數類,但是直接報錯。。有的OJ要求不能有,有的又要求必須有。。
看了一下別人的題解,有兩點收獲:
- 不用判斷整個字符串再確定是什么命令,判斷一下首字母就可以了
- 可以使用
erase
函數整塊刪除。自己就是因為不知道這個函數寫的比較復雜,還是要對STL更加熟悉才行
再研究了一下書上的題解,發現果然pile
函數可以和move
函數合并,而且可是使用resize
函數進行刪除。
對于這種多種指令的,我們要提取出指令之間的共同點,編寫函數以減少重復代碼。
#include <iostream>
#include <vector>
#include <string>
#include <array>using namespace std;namespace {const string QUIT = "quit";const string MOVE = "move";const string ONTO = "onto";const string OVER = "over";const string PILE = "pile";constexpr int MAXN = 25 + 5;array<vector<int>, MAXN> blocks;array<pair<int, int>, MAXN> pos;int n;
}void init() {cin >> n;for (int i = 0; i < n; ++i) {blocks[i].push_back(i);pos[i] = {i, 0};}
}void mve(int x, int idx) {blocks[idx].push_back(x);blocks[pos[x].first].pop_back();pos[x].first = idx;pos[x].second = blocks[idx].size() - 1;
}void pile(int x, int idx) {auto &block_x = blocks[pos[x].first];auto &block_dst = blocks[idx];int origin_pos = pos[x].second;int y;for (int i = origin_pos; i < block_x.size(); ++i) {y = block_x[i];block_dst.push_back(y);pos[y].first = idx;pos[y].second = block_dst.size() - 1;}for (int i = block_x.size() - 1; i >= origin_pos; --i) block_x.pop_back();
}void ret_block(int x) {//return blocks that are stacked on top of xauto &block_x = blocks[pos[x].first];int y;for (int i = block_x.size() - 1; i > pos[x].second; --i) {y = block_x.back();mve(y, y);}
}void work() {int x, y;string action, prep;while (cin >> action) {if (action == QUIT) break;cin >> x >> prep >> y;if (pos[x].first == pos[y].first) continue;if (action == MOVE) {if (prep == ONTO) {//move x onto yret_block(x);ret_block(y);mve(x, pos[y].first);} else {ret_block(x);mve(x, pos[y].first);}} else {if (prep == ONTO) {ret_block(y);pile(x, pos[y].first);} else {pile(x, pos[y].first);}}}
}class Newline {bool first;
public:Newline(bool _fisrt = true):first(_fisrt) {}inline void operator ()();
};inline void Newline::operator()() {if (first) {first = false;} else {cout << "\n";}
}void print() {Newline newline;for (int i = 0; i < n; ++i) {//newline();cout << i << ":";for (auto x : blocks[i]) {cout << " " << x;}cout << "\n";}
}int main() {ios::sync_with_stdio(false);init();work();print();
}