代码源925、全部相等

135 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第1天,点击查看活动详情 logo.png

题目描述

这是代码源5月25日的div2每日一题

全部相等 - 题目 - Daimayuan Online Judge

给定长度为 n 的数组 {A}。

派派非常喜欢 所有元素出现频率相同 的数组,但这样的数组却不常有。派派很伤心 (;´༎ຶД༎ຶ`)。不过聪明的你,发现总能从 {A} 中挑选一个子序列满足上述条件。问此子序列最长为多长?

数据规模

1≤n≤2×105 Ai∈[1,109]

输入格式

输入包含两行,第一行有一个整数 n,表示 {A} 的大小。

接下来一行包含 n 个用空格分隔的整数,依次表示 A1,A2,⋯,An。

输出格式

输出答案。

样例 1 输入

6
1 3 2 1 4 2

样例 1 输出

4

解释:

[1,2,1,2] 满足条件且最长。

问题解析

这题说的是要我们找一个子序列,而且只要子序列内的元素出现频率相等即可,那我们也不用考虑这些元素的先后顺序什么的了。

先把每个数的出现次数记录下来存入数组中并排序,比如样例就会变成:1、1、2、2。

此时问题就变成了:从我们的数组中选出一个数,让(大于等于这个数的个数*这个数)最大即可。

我们从前往后一个个推,因为我们按照升序排序,可以保证后面的数肯定比当前数大,我们拿样例举例来说:

第一个数是1,后面有三个大于等于它的数,加上自己,一共是四个,那么这个子序列的长度就可以是1*4=4。

第二个数是1,后面有两个大于等于它的数,加上自己,一共是三个,那么这个子序列的长度就可以是1*3=3。

第三个数是2,后面有一个大于等于它的数,加上自己,一共是二个,那么这个子序列的长度就可以是2*2=4。

第四个数是2,后面有零个大于等于它的数,加上自己,一共是一个,那么这个子序列的长度就可以是2*1=2。

那么子序列最长是4,可以是【1,2,1,2】,也可以是【1,2,3,4】

AC代码

#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>
#include<math.h>
#include<set>
#include<numeric>
#include<string>
#include<string.h>
#include<iterator>
#include<fstream>
#include<map>
#include<unordered_map>
#include<stack>
#include<list>
#include<queue>
#include<iomanip>

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    vector<int> v(n);
    for (int i = 0; i < n; i++)
    {
        cin >> v[i];
    }
    sort(v.begin(), v.end());
    v.push_back(-1);
    int ans = 1;
    vector<int>cnt;
    for (int i = 1; i <= n; i++)
    {
        if (v[i - 1] != v[i])
        {
            cnt.push_back(ans);
            ans = 1;
        }
        else ans++;
    }
    sort(cnt.begin(), cnt.end());
    n = cnt.size();
    int res = 0;
    for (int i = 0; i < n; i++)
    {
        res = max(res, cnt[i] * (n - i));
    }
    cout << res << endl;
    

    return 0;
}