希望大家收藏:
本文持续更新地址:https://haoqchen.site/2018/09/04/LeetCode841/
今天刷841的时候看到一个大神写的巨高级的代码。在这里分享以下。这位大神用了BFS来解题,然后其中穿插了lambda表达式、stdio与输入输出流相关的知识。读懂后恍然大悟~~~~
该大神的答案:
static auto x = []()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
return 0;
}();
class Solution {
public:
bool canVisitAllRooms(vector<vector<int>>& rooms) {
int room_nums = rooms.size();
list<int> li;
vector<bool> vb(room_nums, false);
int visit_cnt = 0;
li.push_back(0);
++visit_cnt;
vb[0] = true;
while(!li.empty())
{
int idx = li.front();
li.pop_front();
int len = rooms[idx].size();
for(int i = 0; i < len; ++i)
{
if(!vb[rooms[idx][i]])
{
li.push_back(rooms[idx][i]);
vb[rooms[idx][i]] = true;
++visit_cnt;
}
}
}
return visit_cnt == room_nums;
}
};
solution采用BFS来进行解答,相信很多人都能想到,我自己也是这么写的。先定义一个跟rooms长度一样的bool数组vb,初始化为0。如果访问到了一个还未访问的房间(vb为0)则vb相应位置1,并将相应的房间序号放到队列中。对队列进行遍历,最后比较已经访问的房间数跟总房间数是否一直来判断是否都可达。
然而精彩的地方在于:
static auto x = []()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
return 0;
}();
这里作者采用定义变量x的形式让这段代码在main函数前被调用。这里的[](){...}其实是一个lambda表达式,然后x是调用该表达式(后面的();表明其是对表达式的调用),然后用auto做类型声明。同时static表明该变量只用于当前文件,不影响其他文件中的x变量。
然后为什么要这么做呢?因为C++的cin和cout为了兼容C中的scanf、printf(在stdio.h中)函数,让C++代码中可以同时用这四个函数。C++中的输入输出都是先放到缓冲区,再输出。所以效率非常的低,甚至有人做过实验,如果用scanf来进行操作只要2秒,而用cin居然要8秒。ios::sync_with_stdio(false);的作用就是取消iostream的这种兼容。但是需要注意的是,如果取消了这种兼容,C++和C中的输入输出就不能混用了(理论上C++的cin要比scanf快)。
然后cin.tie(nullptr);的作用是取消cin跟cout的绑定。tie是将两个stream绑定的函数,比如可以将输入跟某个文件流指针绑定,就可以直接往文件中输入了。默认cin与cout绑定,解除后可以加快两者的速度,原理没有深入进去尚且不明。