51Nod - 1786 数据流中的算法 - 众数

1786 数据流中的算法 - 众数

基准时间限制：1.5 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

数据流统计功能上线后，为51nod提升用户体验做出了很大的贡献。但是新问题随之而来，夹克老爷还想知道在一个窗口内，访问次数最多用户（即窗口内的众数）。如果有多个众数，取用户ID最小的一个。（窗口的意思是一个固定长度的区间！）

（因为数据流是实时的、在线的，所以不允许使用离线算法^_^）

Input

第一行为整数n, k。(1 <= n <= 5 * 10^6，1 <= k <= 1000）
n代表有多少次操作，k代表窗口大小。

接下来的n行，每行代表一次操作。每行第一个整数为操作数。

操作数1：用户访问
输入格式: <1, id>
用户ID为[0, INT_MAX]闭区间内的整数。代表拥有此ID的用户对网站进行了一次访问，窗口进行相应移动。

操作数2：询问众数
输入格式：<2>
输出窗口内的众数，如果有多个，输出ID最小的那个。

p.s. 对于询问众数的操作，窗口保证不空
p.s.s. 对于询问众数的操作，窗口可能不满

Output

对于询问众数的操作，每行输出一个整数。

Input示例

10 5
1 2
1 1
1 2
1 1 
1 2
1 1
2
1 3
1 3
2

Output示例

1
1

map+set的使用   压入set时要注意排序。

本题思路很简单，但是时间卡的比较紧，网上很多代码都使用了输入输出挂。

如果发现思路没问题但还是超时的话，可以试一下吧iostream的头文件改成stdio.h

输入输出用scanf，测试快1.5秒多。。。。。。

#include<stdio.h>
#include<set>
#include<map>
using namespace std;
struct num
{
    int a,b;
    bool operator < (const num&u) const
    {
        return b==u.b?a<u.a:b>u.b;
    }
};
int p[5000010];
int main()
{
    int n,k;
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        map<int,int> m;
        set<num> s;
        int i,j=0;
        int x,y;
        for(i=0;i<n;i++)
        {
            scanf("%d",&x);
            if(x==1)
            {
                scanf("%d",&y);
                p[++j]=y;
                if(j>k)
                {
                    s.erase(s.find((num){p[j-k],m[p[j-k]]}));
                    m[p[j-k]]--;
                    if(m[p[j-k]])
                        s.insert((num){p[j-k],m[p[j-k]]});
                }
                if(m[y])
                    s.erase(s.find((num){y,m[y]}));
                m[y]++;
                s.insert((num){y,m[y]});
            }
            if(x==2)
            {
                printf("%d\n",(*s.begin()).a);
            }
        }

    }
}