Mad MAD Sum-Codeforces Round 960 (Div. 2)

题目在这里

大意:
MAD函数返回出现次数$\ge 2$的最大整数
$b_i$ = $MAD(a[1,2,...i])$
每次操作把$a_i$进行上述操作，直到全变为0为止，对每次操作的数组进行求和，记为$sum$,问sum的大小

分析:
经过一次运算总可以得到非递减的序列,因为MAD函数非递减，最大值只会越来越大
只有连续的数段可以向右传递,做一次虽然非递减，但是会有只有单个的情况这是不可以向右传递的
我们可以再做一次计算去除这些数，剩下就是可以向右传递的

#include<bits/stdc++.h>
using namespace std;
using i64 = long long;
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);int n;    
void calc(vector<int>&a,i64 &s){map<int,int> mp;i64 now = 0;vector<int> res(n+1,0);for(int i = 1;i<=n;++i){mp[a[i]]++;if(mp[a[i]]>=2&&a[i]>now){now = a[i];}res[i] = now;}a = res;for(int i = 1;i<=n;++i){//cout<<a[i]<<' ';s+=a[i];}//cout<<s<<"\n";
}
void solve(){cin>>n;vector<int>a(n+1);i64 s = 0;for(int i = 1;i<=n;++i) cin>>a[i],s+=a[i];calc(a,s);calc(a,s);for(int i = 1;i<=n;++i){s+=(n-i)*1LL*a[i];}//cout<<s<<"\n";
}signed main(){ios;int t;cin>>t;while(t--){solve();}return 0;
}

读错题目了好难受qaq