您的位置:首页 > 编程语言 > Java开发

java实现找出1000以内的所有完数

2014-08-30 10:35 791 查看
题目意思可以归结为给一个字符串,求可重叠的 k 次最长重复子串。

二分 k ,然后在 height 数组中找是否存大连续 k 个数都大于  k。

代码:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int const N= 1000100;
int wa
, wb
, ws
, wv
;
int rank
, sa
, height
, r
, n, k;

int cmp( int* r, int a, int b, int L ){
return r[a]== r[b] && r[a+ L]== r[b+ L];
}

void da( int* r, int* sa, int n, int m ){
int i, j, p, *x= wa, *y= wb, *t;
for( i= 0; i< m; ++i ) ws[i]= 0;
for( i= 0; i< n; ++i ) ws[ x[i]= r[i] ]++;
for( i= 1; i< m; ++i ) ws[i]+= ws[i-1];
for( i= n- 1; i>= 0; i-- ) sa[ --ws[ x[i] ] ]= i;

for( j= 1, p= 1; p< n; j*= 2, m= p ){
for( p= 0, i= n- j; i< n; ++i ) y[p++]= i;
for( i= 0; i< n; ++i )
if( sa[i]>= j ) y[p++]= sa[i]- j;

for( i= 0; i< n; ++i ) wv[i]= x[y[i]];
for( i= 0; i< m; ++i ) ws[i]= 0;
for( i= 0; i< n; ++i ) ws[ wv[i] ]++;
for( i= 1; i< m; ++i ) ws[i]+= ws[i-1];
for( i= n- 1; i>= 0; i-- ) sa[ --ws[ wv[i] ] ]= y[i];

t= x, x= y, y= t, p= 1; x[ sa[0] ]= 0;
for( i= 1; i< n; ++i )
x[ sa[i] ]= cmp( y, sa[i-1], sa[i], j )? p- 1: p++;
}
}

void callheight( int* r, int*sa, int n ){
int i, j, k= 0;
for( i= 1; i<= n; ++i ) rank[ sa[i] ]= i;

for( i= 0; i< n; height[ rank[i++] ]= k )
for( k?k--:0, j= sa[ rank[i]- 1]; r[i+k]== r[j+k]; k++ );

}

int Check( int mid ){
int ans= 1;
for( int i= 1; i<= n; ++i ){
if( height[i]< mid ) ans= 1;
else ans++;

if( ans>= k ) return 1;
}
return 0;
}

int main(){
scanf("%d%d",&n,&k );
for( int i= 0; i< n; ++i ){
scanf("%d", r+ i ); r[i]++;
}
r
= 0;
da( r, sa, n+ 1, 1000002 );
callheight( r, sa, n );

int left= 1, right= n;
while( left<= right ){
int mid= (left+ right)>>1;

if( Check( mid ) ) left= mid+ 1;
else right= mid- 1;
}
printf("%d\n", right );

return 0;
}

 
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐