[CTCI] 单词最近距离

单词最近距离

题目描述

有一篇文章内含多个单词，现给定两个单词，请设计一个高效算法，找出文中这两个单词的最短距离(即最少相隔的单词数,也就是两个单词在文章中位置的差的绝对值)。

给定一个string数组article，代表所给文章，同时给定文章的单词数n和待查找的两个单词x和y。请返回两个单词的最短距离。保证两个单词均在文中出现且不相同，同时保证文章单词数小于等于1000。

 class Distance {
 public:
     int getDistance(vector<string> article, int n, string x, string y) {
         // write code here
         int px = -, py = -;
         int res = n;
         for (int i = ; i < n; ++i) {
             if (article[i] == x) {
                 px = i;
                 if (py != -) res = min(res, px - py);
             } else if (article[i] == y) {
                 py = i;
                 if (px != -) res = min(res, py - px);
             }
         }
         return res;
     }
 };