牛客多校训练第八场B.Beauty Values(思维)

题目传送门

题意:

输入整数n,再输入n个整数a1,a2,⋯ ,an,表示一段序列,求序列的所有子区间里面不同数字之和。

 

说明:
 在序列1 2 1 3中,
    区间 [1,1],[2,2],[3,3],[4,4]各含1个不同数字
    区间 [1,2],[1,3],[2,3],[3,4]各含2个不同数字
    区间 [1,4],[2,4]各含3个不同数字
 故总和为:1×4+2×4+3×2=18

题解:

枚举每个数字,算出每个数字对答案的贡献,最后把答案累加起来。qq[a]记录a的上一个位置,则a对答案的贡献是(i-qq[a])*(n-i+1)。

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const int maxn=1e5+5;
 5 ll n,a,qq[maxn];
 6 int main()
 7 {
 8     while(~scanf("%d",&n)) {
 9         for(int i=1;i<=n;i++)
10             qq[i]=0;
11         ll ans=0;
12         for(int i=1;i<=n;i++) {
13             scanf("%lld",&a);
14             ans+=(i-qq[a])*(n-i+1);
15             qq[a]=i;
16         }
17         printf("%lld\n",ans);
18     }
19  
20     return 0;
21 }

猜你喜欢

转载自www.cnblogs.com/HOLLAY/p/11333324.html