第九届蓝桥杯省赛B组日志统计

尺取法

一开始程序陷入死循环,没想明白。。。虽然它少操作了不少if语句节省时间,但是边界真的不好处理

这样也是对的

 while(l<len)
    {
        while(r<len&&cnt<k)
        {
            ++cnt;
            ++r;
        }
        if(cnt<k) break;
        else if(cnt>=k)
        {
            if(g[id][r-1]-g[id][l]<d) return true;
            else
            {
                l++;
                cnt--;
            }
        }
    }
#include<iostream>
#include<set>
#include<map>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e5+10;
vector<int> g[N];               //其实用multiset比较好,但是不想写迭代器
vector<int> ans;

int n,d,k;
bool judge(int id)
{
    int cnt=0;
    sort(g[id].begin(),g[id].end());
    int len=g[id].size();
    if(len<k) return false;
    int l=0,r=0;
    while(r<len&&l<=r)              //while(i<=r&&i<len)不行
    {
        while(r<len&&cnt<k)
        {
            ++cnt;
            ++r;                    //这边r++可能到len了,while(i<=r&&i<len)卡不住它 于是乎第二个while进不去,if也进不去,死循环
        }
        if(cnt>=k)
        {
            if(g[id][r-1]-g[id][l]<d) return true;
            else
            {
                l++;
                cnt--;
            }

        }
    }
    return false;

//    while(l<= r&&r< len)
//	{
//		sum++;
//		if(sum>= k)
//		{
//			if(g[id][r]-g[id][l]< d)//注意是小于
//				return 1;
//			else
//				l++,sum--;
//		}
//		r++;
//	}
//    return false;


}
int main()
{
    cin>>n>>d>>k;
    int mx=0;
    for(int i=1;i<=n;++i)
    {
        int ts,id;
        cin>>ts>>id;
        g[id].push_back(ts);
        mx=max(mx,id);
    }
    for(int i=0;i<=mx;++i)
    {
        if(judge(i))
            ans.push_back(i);
    }
    int sum=ans.size();
    for(int i=0;i<sum;++i)
    {
        cout<<ans[i]<<endl;
    }
    return 0;

}

猜你喜欢

转载自blog.csdn.net/iroy33/article/details/88561860