题意:
输入整数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 }