码迷,mamicode.com
首页 > 其他好文 > 详细

hdu 4898 LCP+贪心思维

时间:2014-08-02 15:38:33      阅读:276      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   os   io   for   2014   问题   

题意:将一个字符串切成k块,使得字典序最大的那块最小。

ORZ  WJMZBMR,几行题解读了一天才懂。

快速比较两个子串的大小可以利用LCP(最长公共前缀),比较公共前缀的下一个字符的大小就够了。

利用这种思想,首先我们可以预处理所有子串的LCP(后缀数组+记录 O(2nlog(2n))+O(n*n),dp(O(4*n*n)))

然后将这些子串利用LCP按照字典序排序,开始二分答案。

二分的答案就是这K个块字典序的上限。假设以i作为起点,由于字典序上限已知,所以我们可以立刻求出i点最远能选到哪个点。

现在问题变成了:已知每一个点最远能跳R的距离,求是否存在一条路径,使得跳K次回到起点。

首先我们假设,每个点的R≠0,意思就是每个点都能向后跳,这样我们只需要用贪心的思想,枚举任意点为起点,然后向后能跳多少跳多少,若跳回来所花的次数T<=K,则为true 【由于每个点都能向后跳,则我们一定能够通过改变几个跳跃,使得T==K  若当前的点数<K,也就是每次只跳1步都没法跳到,当然就是false了】

现在的问题就是,如果有某些点R==0怎么办,也就是它一步也不能向后跳,并且其它位置也不能跳到这个位置。所以我们想到,将这个点删除掉,并且,将所有受到影响的点全部减一,以前可以从 a向后跳3步,但现在 b (a<=b<=a+3)被删掉了,所以a只能挑2步了,这样最多迭代n次 处理之后,所有点的R都是不为0的了。

代码写的很挫。。。

#include <cstdio>
#include <iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
#define maxn 2005
char str[maxn];
int sa[maxn],t1[maxn],t2[maxn],c[maxn],n;
void suffix(int m)
{
    int *x=t1,*y=t2;
    for(int i=0; i<m; i++)c[i]=0;
    for(int i=0; i<n; i++)c[x[i]=str[i]]++;
    for(int i=1; i<m; i++)c[i]+=c[i-1];
    for(int i=n-1; i>=0; i--)sa[--c[x[i]]]=i;
    for(int k=1; k<=n; k<<=1)
    {
        int p=0;
        for(int i=n-k; i<n; i++)y[p++]=i;
        for(int i=0; i<n; i++)if(sa[i]>=k)y[p++]=sa[i]-k;
        for(int i=0; i<m; i++)c[i]=0;
        for(int i=0; i<n; i++)c[x[y[i]]]++;
        for(int i=0; i<m; i++)c[i]+=c[i-1];
        for(int i=n-1; i>=0; i--)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;
        x[sa[0]]=0;
        for(int i=1; i<n; i++)
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        if(p>=n)break;
        m=p;
    }
}
int rank[maxn],height[maxn];
void getheight()
{
    int k=0;
    for(int i=0; i<n; i++)rank[sa[i]]=i;
    for(int i=0; i<n; i++)
    {
        if(k)k--;
        if(!rank[i])continue;
        int j=sa[rank[i]-1];
        while(str[i+k]==str[j+k])k++;
        height[rank[i]]=k;
    }
}
int f[maxn][30];
void RMQINIT()
{
    for(int i=0;i<n;i++)
    f[i][0]=height[i];

    for(int j=1;(1<<j)<=n;j++)
        for(int i=0;i+(1<<j)-1<n;i++)
            f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);
}
int RMQ(int l,int r)
{
    if(l>r)swap(l,r);
    l++;
    int k=floor(log(r-l+1.0)/log(2.0));
    return min(f[l][k],f[r+1-(1<<k)][k]);
}
int tn,k,st,mid;
int lcp(int l,int r)
{
    if(l==r) return n-l;
    return RMQ(rank[l],rank[r]);
}
int Lcp[1015][1015];
struct node
{
    int l,r;
    int size;
    bool operator <(const node &x) const
    {
        int LCP=min(size,Lcp[l][x.l]);
        LCP=min(LCP,x.size);
        char a = LCP<size?str[l+LCP]:0;
        char b = LCP<x.size?str[x.l+LCP]:0;
        return a<b;
    }
}a[1111111];
vector<int> v[maxn];
void debug(int pos)
{
    for(int i=a[pos].l;i<a[pos].r;i++)
    {
        putchar(str[i]);
    }
    puts("");
}
bool vis[maxn];
vector<int>far;
bool cal()
{
    far.clear();
    for(int i=0;i<tn;i++)
    {
        if(i==a[mid].l)
        {
            far.push_back(a[mid].size);
            continue;
        }
        int LCP=min(tn,Lcp[i][a[mid].l]);
        LCP=min(LCP,a[mid].size);
        if(LCP>=tn)
        {
            far.push_back(tn);
            continue;
        }
        else
        {
            if(str[i+LCP]<str[a[mid].l+LCP]) far.push_back(tn);
            else far.push_back(LCP);
        }
    }
    int ok=1;
    while(ok)
    {
        ok=0;
        for(int i=0;i<far.size();i++)
        {
            if(far[i]==0)
            {
                for(int j=0;j<far.size();j++)
                {
                    if(j<i&&i<=j+far[j]) far[j]--;
                    else if(j>i&&j+far[j]>=i+far.size()) far[j]--;
                }
                ok=1;
                far.erase(i+far.begin());
            }
        }
    }
    if(far.size()<k) return false;
    int len=far.size();
    for(int i=0;i<len;i++) far.push_back(far[i]);
    for(int i=0;i<len;i++)
    {
        int times=0;
        for(int j=i;j<i+len;j+=far[j]) times++;
        if(times<=k) return true;
    }
    return false;
}
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d%d",&n,&k);
        scanf("%s",str);
        tn=n;
        if(k==1) {puts(str);continue;}
        for(int i=n;i<2*n;i++) str[i]=str[i-n];
        n*=2;
        str[n]=0;
        n++;
        suffix(128);
        getheight();
        RMQINIT();
        n--;
        int top=0;
        for(int i=0;i<tn;i++)
        {
            for(int j=i;j<i+tn;j++)
            {
                a[++top].l=i;
                a[top].r=j+1;
                a[top].size=j+1-i;
            }
        }
        for(int i=0;i<=tn;i++)
            for(int j=0;j<=tn;j++)
                Lcp[i][j]=lcp(i,j);
        sort(a+1,a+top+1);
        int l=1,r=top,ans;
        while(l<=r)
        {
            mid=(l+r)/2;
            if(cal())
            {
                ans=mid;
                r=mid-1;
            }
            else
            {
                l=mid+1;
            }
        }
        debug(ans);
    }
    return 0;
}


hdu 4898 LCP+贪心思维,布布扣,bubuko.com

hdu 4898 LCP+贪心思维

标签:style   blog   color   os   io   for   2014   问题   

原文地址:http://blog.csdn.net/t1019256391/article/details/38347293

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!