《Cracking the Coding Interview》——第18章：难题——题目5

2014-04-29 01:51

题目：你有一个文本文件，每行一个单词。给定两个单词，请找出这两个单词在文件中出现的其中一对位置，使得这两个位置的距离最短。

解法：我的思路是建立倒排索引，计算出所有单词出现的所有位置。下面代码只给出了两个索引的处理方法。倒排索引一般以链表的形式出现，通过顺序扫描两个链表，类似归并排序算法中的merge过程，就能够找出最短距离。请看代码。

代码：

 1 // 18.5 Given a text file containing words, find the shortest distance between two words.

 2 #include <algorithm>

 3 #include <climits>

 4 using namespace std;

 5 

 6 struct ListNode {

 7     int val;

 8     ListNode *next;

 9     ListNode(int _val = 0): val(_val), next(nullptr) {};

10 };

11 

12 class Solution {

13 public:

14     int minDist(ListNode *head1, ListNode *head2)

15     {

16         // the two list records the forward indices of two word, i.e. their occurrence position.

17         // For example, 1->4->5->nullptr means the word appears at 1st, 4th and 5th line.

18         

19         int min_dist = INT_MAX;

20         ListNode *p1, *p2;

21         

22         p1 = head1;

23         p2 = head2;

24         while (p1 != nullptr && p2 != nullptr) {

25             if (p1->val < p2->val) {

26                 min_dist = min(min_dist, p2->val - p1->val);

27                 p1 = p1->next;

28             } else {

29                 min_dist = min(min_dist, p1->val - p2->val);

30                 p2 = p2->next;

31             }

32         }

33         

34         return min_dist;

35     }

36 };