ZOJ 3872 Beauty of Array（模拟）

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
#define LL long long
using namespace std;

int b[1000005];
int main()
{
int n,t,i,j,a;
LL sum,ans;
scanf("%d",&t);
while(t--)
{
sum=ans=0;
memset(b,0,sizeof(b));
scanf("%d",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&a);
if(b[a])
sum-=b[a]*a;
sum+=a*i;
ans+=sum;
b[a]=i;
}
printf("%lld\n",ans);
}
return 0;
}