洛谷P1638 逛画展-最简单的双指针做法

最新推荐文章于 2024-03-08 21:56:34 发布

原创最新推荐文章于 2024-03-08 21:56:34 发布 · 478 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #算法

这篇博客介绍了一种利用双指针算法解决如何在美术馆中选择一个区间，使得该区间内恰好包含指定数量的不同画家的作品，同时使花费的金钱最少。代码中通过遍历画作，动态更新区间，并删除重复画家的作品。最终找到最优的区间范围。

思路：定义双指针x和y代表【x，y】区间，初始化x=y=0开始遍历，刚开始时一直增加y直到包含了所有作家的画，然后就从左边x开始删区间有重复作家的画就行。注意不要把任何一个作家的画删到0！！

代码如下

#include <iostream>
#include <algorithm>
using namespace std;
int arr[1000005];
int co[1000005];//co[i]代表在当前区间第i个作家的画数
int flag;
int main()
{
    int n, m, ans = 0x3f3f3f3f;
    cin >> n >> m;
    int x = 1, ansx, ansy;//区间为【x，y】但y在后面有循环来代替了
    int cnt = 0;//当前区间包含了cnt个作家
    for (int i = 1; i <= n; i++)
    {
        cin >> arr[i];
    }
    for (int i = 1; i <= n; i++)//i代表区间右端y
    {
        if (++co[arr[i]] == 1)//第一次收录该作家的画
            cnt++;
        if (cnt == m)
        {
            while (co[arr[x]] > 1)
            {//如果左边x指的画还有同作者的画那就可以删左边的画
                co[arr[x]]--;
                x++;
            }
            if (ans > i - x)
            {//更新更省钱的方案
                ansy = i;
                ansx = x;
                ans = i - x;
            }
        }
    }
    cout << ansx << " " << ansy << endl;
    return 0;
}