157. Read N Characters Given Read4

最新推荐文章于 2019-09-19 23:44:30 发布

原创最新推荐文章于 2019-09-19 23:44:30 发布 · 163 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

LeetCode 专栏收录该内容

185 篇文章

订阅专栏

本文介绍了一种利用已定义的read4 API实现read函数的方法，该函数可以从文件中读取指定数量的字符。通过实例演示了如何最小化调用read4 API的次数来高效完成读取任务。

https://leetcode.com/problems/read-n-characters-given-read4/description/

The API: int read4(char *buf) reads 4 characters at a time from a file.

The return value is the actual number of characters read. For example, it returns 3 if there is only 3 characters left in the file.

By using the read4 API, implement the function int read(char *buf, int n) that reads n characters from the file.

Example 1:

Input: buf = "abc", n = 4
Output: "abc"
Explanation: The actual number of characters read is 3, which is "abc".

Example 2:

Input: buf = "abcde", n = 5 
Output: "abcde"

Note:
The read function will only be called once for each test case.

# The read4 API is already defined for you.
# @param buf, a list of characters
# @return an integer
# def read4(buf):

class Solution(object):
    def read(self, buf, n):
        buf4 = [""] * 4  # read4 buffer 
        count = 0

        while n > 0:
            now = min(n, read4(buf4))
            if now == 0:  # no more to read
                break
            buf[count:count+now+1] = buf4  # copy from buf4 to buf
            count += now
            n -= now
        return count