D. Cut and Stick
(赛后补题)借本题学习莫队算法以及区间众数的求法
题意:对于整型数组,每次询问 [ L , R ] [L,R] [L,R]区间问最少分为多少个子序列,使得每个子序列的众数 x x x的个数 c n t x cnt_x cntx不大于 ⌈ l e n 2 ⌉ \left \lceil \frac{len}{2} \right \rceil ⌈2len⌉, l e n len len表示子序列的长度
思路:对于每次询问只需知道询问区间内的众数
x
x
x的个数即可,最优解即为将其余的非众数尽可能与更多的
x
x
x组合为一个子序列,而剩下的
x
x
x每个自为一个子序列
给一个不严格的证明,图中有
a
a
a个众数与
b
b
b个非众数,法①将全部的非众数放在同一个子序列中,此时该子序列最多匹配
b
+
1
b+1
b+1个众数,而剩余的众数则需单独出现;法②将b个非众数分为两部分(
c
+
d
=
b
c+d=b
c+d=b),两部分分别匹配
c
+
1
c+1
c+1和
d
+
1
d+1
d+1个众数(
c
+
1
+
d
+
1
=
c
+
d
+
2
=
b
+
2
c+1+d+1 = c+d+2 = b+2
c+1+d+1=c+d+2=b+2)此时比法①多匹配了一个众数,然而由于被分为两部分,自然地也多产生了一个子序列,因此两者是等效的,可见只要将尽可能多的众数匹配给非众数,无论非众数分为多少块其效果是等效的
因此问题便转化为求区间众数的个数问题。
数组长度:
n
n
n,询问次数:
m
m
m
若采用暴力求解,每次询问计算众数的个数的时间复杂度为
O
(
n
)
O(n)
O(n),总时间复杂度为
O
(
m
∗
n
)
O(m*n)
O(m∗n),会超时
采用莫队算法优化,可将每次询问的平均时间复杂度降为
O
(
n
)
O(\sqrt{n})
O(n)
莫队步骤:
- 将数组分为大小为 n \sqrt{n} n的块
- 将询问顺序按下面的规则排序
首先按照 L L L所在的块号排序,块号越小越靠前(升序)
若 L L L所在的块号相同,则按 R R R所在的块号升序 - 接着定义两个函数 a d d ( ) add() add()和 d e l ( ) del() del()分别用于添加和删除元素,用于在上一个询问的区间基础上增删元素得到当前询问的区间,每次调用增删函数的关键是动态地更新众数的状态
维护众数状态:
采用
m
a
p
map
map映射存储每个数出现的次数,采用数组
a
a
a记录有多少个数出现了
i
i
i次,当前的众数的个数即为数组
a
[
i
]
>
0
a[i]>0
a[i]>0的最大的
i
i
i,每次增删实时更新两者的数据即可
#include<bits/stdc++.h>
#define debug1(x) cout<<#x<<":"<<x<<endl
#define fastio() ios_base::sync_with_stdio(0);cin.tie(0)
typedef long long ll;
typedef unsigned long long ull;
const int N = (int)3e5+100;
const int RN = sqrt(N)+10;
const int MOD = 10000;
using namespace std;
int a[N];
map<int,int> mp[RN];
int n,m,block;
int cnt[N],sum[N],ans[N],max_cnt = 0;
struct query
{
int id;
int l;
int r;
}q[N];
bool cmp(query a, query b)
{
int al = a.l / block, bl = b.l / block;
int ar = a.r / block, br = b.r / block;
if(al != bl) return al < bl;
return ar < br;
}
void add(int e)
{
sum[cnt[e]]--;
cnt[e]++;
sum[cnt[e]]++;
max_cnt = max(max_cnt, cnt[e]);
}
void del(int e)
{
sum[cnt[e]]--;
if(cnt[e] == max_cnt && sum[cnt[e]] == 0) max_cnt--;
cnt[e]--;
sum[cnt[e]]++;
}
int main()
{
cin>>n>>m;
block = sqrt(n);
for(int i = 1; i <= n; i++) scanf("%d",&a[i]);
int l, r;
for(int i = 0; i < m; i++)
{
scanf("%d%d",&l,&r);
q[i].id = i;
q[i].l = l;
q[i].r = r;
}
sort(q, q + m, cmp);
int cl, cr, width;
cl = cr = 0; add(a[0]);
for(int i = 0; i < m; i++)
{
l = q[i].l, r = q[i].r;
while(cr < r) add(a[++cr]);
while(cl > l) add(a[--cl]);
while(cr > r) del(a[cr--]);
while(cl < l) del(a[cl++]);
width = r - l + 1;
ans[q[i].id] = max(2 * max_cnt - width, 1);
}
for(int i = 0; i < m; i++) printf("%d\n", ans[i]);
return 0;
}