Zoj 3842 Beauty of Array

Problem地址：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=5520

根据题目的要求，需要算出所有连续子数组的the beauty的总和。

那么要求这个这个总和，刚开始最容易想到的就是这样：

for( int i=1; i<=N; i++ ) {

    for( int j = 1; j<=i; j++ ) {

         ... //排除重复的数计算总和   

    }       

}

 这样子的结果实际上是 Time Limit Exceeded

因此采取这种方法是不当的，那么继续思考：

先思考没有重复的情况

例如：1，2 可以有子数组 1（1）， 1，2（2），2（2）,三个子数组，括号内为该子数组的beauty，

如果后面在加上一个3呢？可以得到子数组1（1）， 1，2（2），2（2），1，2，3（6），2，3（5），3（3），可见增加了3个3，之所以是3个3。是因为这个3处在第三个位置。

同理，如果再增加一个数，形成数组列1，2，3，4，那么和1，2，3相比，其总和应该增加4个4

可以得出，没有重复的长度为n的数组，如果在第n+1个位置，再增加一个不重复的数x，则与没有重复的长度为n的数组的总beauty要增加x*(n+1)

 

既然得出了这一点，那么考虑存在重复的情况：

在1，2，3，2，到第四个数，即2时，新增的序列为1,2,3,2   ,   2,3,2,  3,2,  2,但是和前一个3相比，以这个2为结尾的序列总beauty值增加了2*2，因为这个2与上一个2距离为2.

 

最后可以得出，每读进一个数，就可以根据前一个数算出以这个数为结尾的数列的beauty值。可以使默认每一个数的初始位置为0，之后进行更新。

最后计算总和就行了。

 

#include <iostream>

#include <cstring>

#include <cstdio>



using namespace std;



const int NUM_MAXN = 1000000 + 50; // the max number

long long value[ NUM_MAXN ]; // the sub ending with i position can totally get value[i] beauty

// Attention : long long

int pos[ NUM_MAXN ]; // the number i in the pos[i] position



int main() {

	int T;

	cin >> T;

	while( T -- ) {

		int N;

		scanf( "%d", &N );

		memset( pos, 0, sizeof(pos) );

		value[ 0 ] = 0;

		int tmp;

		for( int i=1; i<=N; i++ ) {

			scanf( "%d", &tmp );

			value[i] = value[ i-1 ] + ( i-pos[ tmp ] ) * tmp;

			pos[ tmp ] = i;

		}

		long long sum = 0;

		// Attention : long long

		for( int i=1; i<=N; i++ ) {

			sum = sum + value[ i ];

		}

		// Attention : long long

		printf( "%lld\n", sum );

	}

	return 0;

}