Lintcode算法：1823. 数组的最长前缀--打败 97.09% 的提交!

1.题目：

给定两个正整数X和Y，以及正整数数组nums。
我们需要找到一个最大的index，使得在nums[0], nums[1], … , nums[index]中，出现X、Y的次数相等，且至少均出现一次，返回该index。
若不存在这样的index，则返回-1。
样例
示例 1:
输入:
X = 2
Y = 4
nums: [1, 2, 3, 4, 4, 3]
输出: 3
解释: 保证 2 和 4 出现相同次数的最长前缀是： {1, 2, 3, 4}，所以你应该返回3。
示例 2：
输入：
X = 7
Y = 42
nums = [7、42、5、6、42、8、7、5、3、6、7]
输出：9
解释：保证7和42出现相同次数的最长前缀是：{7, 42, 5, 6, 42, 8, 7, 5, 3, 6}，所以你应该返回9。
示例 3:
输入:
X = 1
Y = 10
nums: [2, 3, 1]
输出：-1
解释：不存在前缀使得 1 和 10 都出现且出现次数相同的情况

2.思路

遍历给出的nums，分别将X，Y出现的位置记录在两个vector内，对比两个vector的长度al、bl，分情况考虑即可确定index的值：

代码

class Solution {
public:
    /**
     * @param X: a integer
     * @param Y: a integer
     * @param nums: a list of integer
     * @return: return the maximum index of largest prefix
     */
    int LongestPrefix(int X, int Y, vector<int> &nums) {
        // write your code here
        int k=0,l=nums.size();
        vector<int> a,b;
        //记录X，Y出现的位置，分别存入a，b
        for(int i=0;i<l;i++){
            if(nums[i]==X){
                a.push_back(i);
                k++;
            }else if(nums[i]==Y){
                b.push_back(i);
                k--;
            }
        }
        int al=a.size(),bl=b.size();
        if(al==0||bl==0){
            return -1;
        }
        if(al==bl){
            return l-1;
        }else if(al>bl){
            while(a[bl]<b[bl-1]){
                b.pop_back();
                bl--;
                if(bl==0) return -1;
            }
            return a[bl]-1;
        }else{
            while(b[al]<a[al-1]){
                a.pop_back();
                al--;
                if(al==0) return -1;
            }
            return b[al]-1;
        }
    }
};

结果