hdu--4455+ Substrings+2012杭州區域賽C題+DP

acm_lkl發表於2020-04-04

好久沒寫部落格了,看了下上一篇部落格還是八月份的最後一天寫的,然後就因為阿里擁抱變化,忙著找工作去了;9月份找到工作,10月份又胡思亂想了一個月,但其實都是沒必要的,應該明白鬍思亂想解決不了任何問題的,實在得不到的又何必強求,順其自然吧。加油吧,少年!!!

題目連結:點選進入
下午訓練的時候這道題想了好幾個小時還是完全沒有思路,但是靜下心來還是可以看到一些規律的。首先w=i+1的情況和w=i時的情況是有關係的,具體的我們設dp[i]表示w=i時的答案,那麼考慮到w=i-1時,如果在每一段的後面加上一個數,就可以得到w=i的情況;但是要注意的是w=i-1時的最後一段是不能通過加上一個數得到w=i的情況的,所以這種情況要去掉,這種情況的數目是很容易計算出來的,因為只需要固定起點為最後一個元素,然後依次向前推看當前區間有多少個不同的元素;然後難點在於如何計算在w=i-1情況下,對每一段加上一個數後增加的情況數;對於這一點我們可以提前預處理出每個數和其最近的相同數的距離來計算(如果沒有相同的數,就是和原點的距離),我們統計出各種不同距離的情況數後,就可以利用遞推的方法計算出w=i-1時通過在每一段加一個數後增加的情況數了。更具體的見程式碼:

程式碼如下:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

const int maxn=1000010;
long long dp[maxn];
int sum[maxn],pre[maxn],dist[maxn];
int last[maxn],a[maxn];

int main()
{
    //freopen("in.txt","r",stdin);
    int n;
    while(scanf("%d",&n)&&n)
    {
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        memset(pre,0,sizeof(pre));
        last[1]=1;
        pre[a[n]]=1;
        int cnt=2;
        for(int i=n-1;i>=1;i--)
        {
            if(pre[a[i]])
                last[cnt]=last[cnt-1];
            else
            {
                last[cnt]=last[cnt-1]+1;
                pre[a[i]]=1;
            }
            cnt++;
        }
        memset(pre,0,sizeof(pre));
        memset(dist,0,sizeof(dist));
        for(int i=1;i<=n;i++)
        {
            dist[i-pre[a[i]]]++;
            pre[a[i]]=i;
        }
        sum[n]=dist[n];
        for(int i=n-1;i>=1;i--)
        {
            sum[i]=sum[i+1]+dist[i];
        }
        dp[1]=n;
        for(int i=2;i<=n;i++)
            dp[i]=dp[i-1]-last[i-1]+sum[i];
        int m;
        scanf("%d",&m);
        while(m--)
        {
            int x;
            scanf("%d",&x);
            printf("%lld\n",dp[x]);
        }
    }
  return 0;
}

相關文章