#P11390. [COCI 2024/2025 #1] 教师 / Učiteljica

    ID: 12750 远端评测题 5000ms 500MiB 尝试: 0 已通过: 0 显示难度省选/NOI− 上传者: 标签>线段树2024容斥原理COCI(克罗地亚)

[COCI 2024/2025 #1] 教师 / Učiteljica

背景

译自 COCI 2024/2025 #1 T4。5s,0.5G\texttt{5s,0.5G}。满分为 120120。

题目描述

给定长度为 nn 的正整数序列 a1,a2,⋯ ,ana_1,a_2,\cdots,a_n。给定常数 kk。

求出满足以下条件的二元组 (l,r)(l,r) 的数量:

  • 1≤l≤r≤n1\le l\le r\le n;
  • 对于任意 1≤i≤k1\le i\le k,都存在一个数 xx,使得 xx 在 al,al+1,…,ara_l,a_{l+1},\ldots,a_r 间出现恰好 ii 次。

输入格式

第一行,两个正整数 n,kn,k。

第二行,nn 个正整数 a1,a2,⋯ ,ana_1,a_2,\cdots,a_n。

输出格式

输出一行一个整数,表示答案。

3 1
1 2 1
6
6 3
6 5 6 4 5 5
1
6 2
5 4 5 2 6 5
5

提示

对于 100%100\% 的数据,保证:

  • 1≤n≤1051\le n\le 10^5;
  • 1≤k≤41\le k\le 4;
  • 1≤ai≤n1\le a_i\le n。
子任务编号 n≤n\le 特殊性质 得分
1 1 10310^3 20 20
2 2 10510^5 A 15 15
3 3 B 35 35
4 4 50 50
  • 特殊性质 A:1≤ai≤k1\le a_i\le k。
  • 特殊性质 B:k=1k=1。