CF 190D. Non-Secret Cypher
2016-07-18 19:17
253 查看
D. Non-Secret Cypher
time limit per test
3 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
Berland starts to seize the initiative on the war with Flatland. To drive the enemy from their native land, the berlanders need to know exactly how many more flatland soldiers are left in the enemy's reserve. Fortunately, the scouts captured an enemy in the
morning, who had a secret encrypted message with the information the berlanders needed so much.
The captured enemy had an array of positive integers. Berland intelligence have long been aware of the flatland code: to convey the message, which contained a number m, the enemies use an array of integers a.
The number of its subarrays, in which there are at least k equal numbers, equals m. The number k has long been known in the Berland army so
General Touristov has once again asked Corporal Vasya to perform a simple task: to decipher the flatlanders' message.
Help Vasya, given an array of integers a and number k, find the number of subarrays of the array of numbers a, which has at least k equal
numbers.
Subarray a[i... j] (1 ≤ i ≤ j ≤ n) of array a = (a1, a2, ..., an) is
an array, made from its consecutive elements, starting from the i-th one and ending with the j-th one: a[i... j] = (ai, ai + 1, ..., aj).
Input
The first line contains two space-separated integers n, k (1 ≤ k ≤ n ≤ 4·105), showing how many numbers
an array has and how many equal numbers the subarrays are required to have, correspondingly.
The second line contains n space-separated integers ai (1 ≤ ai ≤ 109)
— elements of the array.
Output
Print the single number — the number of such subarrays of array a, that they have at least k equal integers.
Please do not use the %lld specifier to read or write 64-bit integers in С++. In is preferred to use the cin, cout streams or the %I64d specifier.
Examples
Input
Output
Input
Output
Input
Output
Note
In the first sample are three subarrays, containing at least two equal numbers: (1,2,1), (2,1,2) and (1,2,1,2).
In the second sample are two subarrays, containing three equal numbers: (1,2,1,1,3) and (1,2,1,1).
In the third sample any subarray contains at least one 1 number. Overall they are 6: (1), (1), (1), (1,1), (1,1) and (1,1,1).
题意:给出n个元素的数列(n<=4e5),问有多少个子区间满足至少有一个数值出现至少K次。
解法1:
从小到大,枚举区间左端点,找对应的最左区间右端点,可以发现最左区间右端点满足单调不减,所以算法的时间复杂度是O(2*n)=O(n)
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;
const int maxn=4*100000 ;
int n,K;
int a[maxn+10];
map<int,int >mp;
ll solve()
{
int le=1,ri=0;
bool ok=false;
ll ans=0;
while(ri<=n)
{
if(ok){
ans+=n-ri+1;
if(--mp[a[le++]]==K-1) ok=false;
}
else {
if( ++mp[a[++ri]]==K ) ok=true;
}
}
return ans;
}
int main()
{
while(~scanf("%d%d",&n,&K))
{
mp.clear();
for1(i,n)
{
scanf("%d",&a[i]);
}
printf("%I64d\n",solve() );
}
return 0;
}
解法二:
对应于解法一,枚举区间右端点,可是最右左端点并不是单调不减的。
先对所有数值离散化。记录每个值x出现了几次,并且第k次出现的位置。
右端点为i时,考虑求其最右左端点R[i](假设存在),
如果a[i]的值为x,现在x出现了m次,考虑第m-K+1个x出现的位置pos。
如果R[i-1]存在,那么R[i]=max(R[i-1],pos);
ans=∑(R[i])
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;
const int maxn= 400000 ;
map<int ,int >mp;
int a[maxn+10];
vector<int >pos[maxn+10];
int num[maxn+10],n,N,K;
int R[maxn+10];
int ID(int x)
{
if(mp.count(x))
{
return mp[x];
}
int k=mp.size();
mp[x]=k;
return mp[x];
}
void work()
{
ll ans=0;
R[0]=INF;
N=mp.size();
for0(i,N)
{
num[i]=0;
pos[i].clear();
}
for1(i,n)
{
int x=a[i];
pos[x].push_back(i);
num[x]++;
if(num[x]<K)
{
if(R[i-1]==INF)
{
R[i]=INF;
continue;
}
R[i]=R[i-1];
ans+=R[i];
continue;
}
int p=num[x]-K+1-1;
p=pos[x][p];
if(R[i-1]==INF) R[i]=p;
else R[i]=max(p,R[i-1]);
ans+=R[i];
}
printf("%lld\n",ans);
}
int main()
{
while(~scanf("%d%d",&n,&K))
{
mp.clear();
int x;
for1(i,n)
{
scanf("%d",&x);
a[i]=ID(x);
}
work();
}
return 0;
}
time limit per test
3 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
Berland starts to seize the initiative on the war with Flatland. To drive the enemy from their native land, the berlanders need to know exactly how many more flatland soldiers are left in the enemy's reserve. Fortunately, the scouts captured an enemy in the
morning, who had a secret encrypted message with the information the berlanders needed so much.
The captured enemy had an array of positive integers. Berland intelligence have long been aware of the flatland code: to convey the message, which contained a number m, the enemies use an array of integers a.
The number of its subarrays, in which there are at least k equal numbers, equals m. The number k has long been known in the Berland army so
General Touristov has once again asked Corporal Vasya to perform a simple task: to decipher the flatlanders' message.
Help Vasya, given an array of integers a and number k, find the number of subarrays of the array of numbers a, which has at least k equal
numbers.
Subarray a[i... j] (1 ≤ i ≤ j ≤ n) of array a = (a1, a2, ..., an) is
an array, made from its consecutive elements, starting from the i-th one and ending with the j-th one: a[i... j] = (ai, ai + 1, ..., aj).
Input
The first line contains two space-separated integers n, k (1 ≤ k ≤ n ≤ 4·105), showing how many numbers
an array has and how many equal numbers the subarrays are required to have, correspondingly.
The second line contains n space-separated integers ai (1 ≤ ai ≤ 109)
— elements of the array.
Output
Print the single number — the number of such subarrays of array a, that they have at least k equal integers.
Please do not use the %lld specifier to read or write 64-bit integers in С++. In is preferred to use the cin, cout streams or the %I64d specifier.
Examples
Input
4 2 1 2 1 2
Output
3
Input
5 3 1 2 1 1 3
Output
2
Input
3 1 1 1 1
Output
6
Note
In the first sample are three subarrays, containing at least two equal numbers: (1,2,1), (2,1,2) and (1,2,1,2).
In the second sample are two subarrays, containing three equal numbers: (1,2,1,1,3) and (1,2,1,1).
In the third sample any subarray contains at least one 1 number. Overall they are 6: (1), (1), (1), (1,1), (1,1) and (1,1,1).
题意:给出n个元素的数列(n<=4e5),问有多少个子区间满足至少有一个数值出现至少K次。
解法1:
从小到大,枚举区间左端点,找对应的最左区间右端点,可以发现最左区间右端点满足单调不减,所以算法的时间复杂度是O(2*n)=O(n)
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;
const int maxn=4*100000 ;
int n,K;
int a[maxn+10];
map<int,int >mp;
ll solve()
{
int le=1,ri=0;
bool ok=false;
ll ans=0;
while(ri<=n)
{
if(ok){
ans+=n-ri+1;
if(--mp[a[le++]]==K-1) ok=false;
}
else {
if( ++mp[a[++ri]]==K ) ok=true;
}
}
return ans;
}
int main()
{
while(~scanf("%d%d",&n,&K))
{
mp.clear();
for1(i,n)
{
scanf("%d",&a[i]);
}
printf("%I64d\n",solve() );
}
return 0;
}
解法二:
对应于解法一,枚举区间右端点,可是最右左端点并不是单调不减的。
先对所有数值离散化。记录每个值x出现了几次,并且第k次出现的位置。
右端点为i时,考虑求其最右左端点R[i](假设存在),
如果a[i]的值为x,现在x出现了m次,考虑第m-K+1个x出现的位置pos。
如果R[i-1]存在,那么R[i]=max(R[i-1],pos);
ans=∑(R[i])
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
using namespace std;
#define all(x) (x).begin(), (x).end()
#define for0(a, n) for (int (a) = 0; (a) < (n); (a)++)
#define for1(a, n) for (int (a) = 1; (a) <= (n); (a)++)
typedef long long ll;
typedef pair<int, int> pii;
const int INF =0x3f3f3f3f;
const int maxn= 400000 ;
map<int ,int >mp;
int a[maxn+10];
vector<int >pos[maxn+10];
int num[maxn+10],n,N,K;
int R[maxn+10];
int ID(int x)
{
if(mp.count(x))
{
return mp[x];
}
int k=mp.size();
mp[x]=k;
return mp[x];
}
void work()
{
ll ans=0;
R[0]=INF;
N=mp.size();
for0(i,N)
{
num[i]=0;
pos[i].clear();
}
for1(i,n)
{
int x=a[i];
pos[x].push_back(i);
num[x]++;
if(num[x]<K)
{
if(R[i-1]==INF)
{
R[i]=INF;
continue;
}
R[i]=R[i-1];
ans+=R[i];
continue;
}
int p=num[x]-K+1-1;
p=pos[x][p];
if(R[i-1]==INF) R[i]=p;
else R[i]=max(p,R[i-1]);
ans+=R[i];
}
printf("%lld\n",ans);
}
int main()
{
while(~scanf("%d%d",&n,&K))
{
mp.clear();
int x;
for1(i,n)
{
scanf("%d",&x);
a[i]=ID(x);
}
work();
}
return 0;
}
相关文章推荐
- oracle数据库如何备份一张表
- HDU 5718 Oracle
- 设计原则6大原则解读--
- 如何在 Linux 上永久挂载一个 Windows 共享
- ACM知识点 之 贪心(1)理论初探
- 转圈游戏
- [置顶] 不定时整理的百度电子书(2016 10 8 更新)
- 反向传播神经网络(有代码)
- Xlib.h No such file or directory 问题解决
- cocos creator配置表的添加
- 问题记录:hadoop集群提交job时出现Exception in thread "main" java.io.IOException: Error opening job jar:
- 【Creator】学习笔记
- DAY7——液晶显示
- linux c语言的格式化输入输出
- Hibernate总结(四)--session的产生方式和一对多的关系映射
- h5 video
- 模板特化和偏特化
- Angularjs之移动APP开发
- Android图片异步加载框架Android-Universal-Image-Loader
- 拖放事件原理及实现详解