《Cracking the Coding Interview》——第13章：C和C++——题目1

2014-04-25 19:13

题目：用C++写一个读取文件倒数K行的方法。

解法：因为是要取倒数K行，所以我的思路是一行一行地读。过程中需要保存一个长度为K的链表，每次新读到一行都将表头节点移到表尾，如此滚动直到文件结束。最后表头节点中存的字符串就是倒数第K行，依次输出就是倒数K行了。不过当时做这题的时候貌似偷懒了，直接用python写了个偷懒并且不高效的方法。

代码：

import sys
import os
import os.path

if __name__ == '__main__':
    if len(sys.argv) == 1 or len(sys.argv) > 3:
        print "Usage: python p13-1.py [FILE NAME] [k]"
        print "Description: output the last k lines of a file."
        sys.exit(0);
    if len(sys.argv) == 2:
        k = 5
    if len(sys.argv) == 3:
        k = int(sys.argv[2])
        k = 5 if k <= 0 else k
    file_name = sys.argv[1]
    if os.path.isfile(file_name) == False:
        print "File \"%s\" is not found." % file_name
        sys.exit(1)
    file = open(file_name, 'r')
    lines = file.readlines()
    file.close()
    k = len(lines) if k > len(lines) else k
    line_count = len(lines)
    for i in range(line_count - k, line_count):
        print lines[i],