E - Make it Palindrome （abc290）(字符串)

E - Make it Palindrome (atcoder.jp)

给定一个字符串，求每个子串要变成回文串的总的最小花费，

#include <iostream>
#include <bits/stdc++.h>
#include <cstring>
#include <vector>
#include <set>
#include <queue>
#include <algorithm>
#define x first
#define y second
#define pb emplace_back
#define fu(i,a,b) for(int i=a;i<=b;i ++ )
#define fd(i,a,b) for(int i=a;i>=b;i -- )
#define endl '\n'
#define ms(x,y) memset(x,y,sizeof x)
#define ios ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
using namespace std;

typedef long long LL;
typedef vector<vector<LL>> VVL;
typedef vector<LL> VL;
typedef vector<int> VI;
typedef pair<int,int> PII;
typedef pair<PII,int> PIII;
typedef pair<double,double> PDD;
typedef pair<double,int> PDI;
typedef pair<char,int> PCI;
typedef pair<string,int> PSI;
typedef pair<int,string> PIS;
typedef pair<LL,LL> PLL;
typedef __int128 i128;
typedef unsigned long long ULL;
const int N = 3e5+ 10,M = N *4  ,INF = 0x3f3f3f3f ;
const int mod = 1e9; 
const double eps = 1e-8;




inline void solve()
{
    int n ;cin >> n;
    
    VI a(n),cnt(n);
    fu(i,0,n-1) cin >> a[i],a[i] -- ;
    LL ans=0,sum=0;
    int tot = 0; // 当前已经考虑的串的长度 
    
    
//  每次取前面和 后面的子串 
// 先考虑后面的子串长度多1的奇数串，然后在考虑前后长度一样的偶数串
    fd(i,(n-1)/2,0)
    {
        sum += tot - cnt[a[i]] ++ ;
// 与当前的数不同的数的个数 ，都会答案有多次影响 
        tot ++ ;
        if(i  != n - 1 -i  ) 
    // 不是长度为奇数的串的中间位置 
        {
            sum += tot - cnt[a[n-1-i]] ++ ;
            tot ++ ;
        }
        ans += sum;
    }
    cout << ans << endl;
        
    
    
}
signed main()
{
//  freopen("1.txt","w",stdout);
    ios

    int t=1;
//    cin>>t;
    int now = 1;
    while(t -- )
    {
//      cout<<"Case "; 
//      cout<<"Scenario #"; 
//      cout<< now ++ <<": ";
//      cout<< now ++ <<": \n";
        solve();
    }


    return 0;
}