zoj 3872(dp)

//題意:求一段序列裏有多個連續的子序列 每個連續子序列中各個元素(相同的只算一個)相加之和爲n 
//           求所有這樣的n累加之和
//方法:動規的思想 dp[i]表示以第i個元素爲右頂點的所有區間的累加和
//           利用pre數組 pre[i] 記錄的是元素i上一次出現的位置
//           ss數組保存這段序列
//
//狀態轉移方程:dp[i] = dp[i - 1] + (i - pre[ss[i]]) * ss[i];
//                   
#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <iostream>
using namespace std;
int pre[1000005];
int main()
{
    int t;
    int ss[100005];
    long long int dp[100005], ans;

    scanf("%d", &t);
    while(t--)
    {
        int n;
        scanf("%d", &n);
        for(int i = 0; i < n; i++)
            scanf("%d", &ss[i]);
        ans = 0;
        memset(pre, -1, sizeof pre);
        dp[0] = ss[0];
        pre[ss[0]] = 0;
        for(int i = 1; i < n; i++)
        {
            dp[i] = dp[i - 1] + 1ll * (i - pre[ss[i]] ) * ss[i];
            pre[ss[i]] = i;
            ans += dp[i];
        }
        printf("%lld\n", ans + dp[0]);
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章