【DP】ZOJ - 3872 Beauty of Array

Step1 Problem:

定义Beauty数是一个序列中所有不同数的和，求一个序列所有连续子序列的Beauty和
数据范围：
1<=N<=1e5, 0<=序列值<=1e6

Step2 Involving algorithms:

DP

Step3 Ideas:

状态dp[i] : 以第i个数结尾的子序列的和
pos[x]：按输入顺序，x最后出现的位置
例如：
1 2 3 3
i = 1的时候：1
i = 2的时候：1 2，2
i = 3的时候：1 2 3，2 3，3
i = 4的时候：1 2 3 3，2 3 3，3 3，3
所以
dp[i] = dp[i-1] + (i-pos[x])*a[i];
ans = dp[1] + dp[2] + … + dp[n];

Step4 Code:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e6+100;
ll dp[N];
int pos[N];
int main()
{
    int T, n;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%d", &n);
        memset(dp, 0, sizeof(dp));
        memset(pos, 0, sizeof(pos));
        ll sum = 0, num;
        for(int i = 1; i <= n; i++)
        {
            cin >> num;
            dp[i] = dp[i-1] + (i - pos[num])*num;
            sum += dp[i];
            pos[num] = i;
        }
        cout << sum << endl;
    }
    return 0;
}