题目链接:http://codeforces.com/problemset/problem/190/D
题目大意:求一个数列中有多少子数列其中某个数出现k次以上.
题目思路:枚举每个数,求出以当前数为结尾可以获得的符合要求的子数列有几个,全部加起来就是答案了(= =,感觉就是废话,那我为嘛打?..........)
关键在于如果获取"以当前数为结尾可以获得的符合要求的子数列有几个",首先我们可以用vector<>pos[i]来保存每个数出现的位置,i表示这个数(离散后.),还需要一个cnt[i]表示i在当前位置出现了几次.
num[cur]=max(pos[i][cnt[i]-k])+1 (i<cur,pos[i][cnt[i]-k]表示起始位置离cur最近的最短合法字串的起始位置)
代码:
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <time.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;
//#define ull unsigned __int64
#define ll __int64
//#define ull unsigned long long
//#define ll long long
#define son1 New(p.xl,xm,p.yl,ym),(rt<<2)-2
#define son2 New(p.xl,xm,min(ym+1,p.yr),p.yr),(rt<<2)-1
#define son3 New(min(xm+1,p.xr),p.xr,p.yl,ym),rt<<2
#define son4 New(min(xm+1,p.xr),p.xr,min(ym+1,p.yr),p.yr),rt<<2|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define middle (l+r)>>1
#define MOD 1000000007
#define esp (1e-8)
const int INF=0x3F3F3F3F;
const double DINF=10000.00;
//const double pi=acos(-1.0);
const int N=400010;
int n,m;
int cnt[N];
vector<int>pos[N];
ll hash[N],a[N];
int bs(ll key,int size,ll A[]){
int l=0,r=size-1,mid;
while(l<=r){
mid=middle;
if(key>A[mid]) l=mid+1;
else if(key<A[mid]) r=mid-1;
else return mid;
}return -1;
}
int main(){
//freopen("1.in","r",stdin);
//freopen("1.out","w",stdout);
int i,j,k;
ll p,r;
//int T,cas;scanf("%d",&T);for(cas=1;cas<=T;cas++){
while(~scanf("%d%d",&n,&k)){
for(i=0;i<n;i++) scanf("%I64d",&a[i]),hash[i]=a[i],pos[i].clear();
sort(hash,hash+n);
for(i=m=1;i<n;i++) if(hash[i]!=hash[i-1]) hash[m++]=hash[i];
memset(cnt,0,sizeof(cnt));
for(i=j=0,r=p=0;i<n;i++){
j=bs(a[i],m,hash);
cnt[j]++;
pos[j].push_back(i);
if(cnt[j]>=k) p=max(p,(ll)pos[j][cnt[j]-k]+1);//,printf("**%I64d %I64d**\n",a[i],p);
r+=p;
}
printf("%I64d\n",r);
}
return 0;
}