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;
}