結果

問題 No.2938 Sigma Sigma Distance Distance Problem
ユーザー 沙耶花
提出日時 2024-10-18 21:30:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,012 bytes
コンパイル時間 4,130 ms
コンパイル使用メモリ 252,056 KB
最終ジャッジ日時 2025-02-24 20:28:12
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000000LL

int main(){
	
	int n;
	cin>>n;
	
	vector<vector<int>> a(501);
	rep(i,n){
		int t;
		cin>>t;
		a[t].push_back(i);
	}
	long long ans = 0;
	rep(i,501){
		for(int j=i+1;j<501;j++){
			if(a[i].empty() || a[j].empty()) continue;
			long long tsum = 0;
			{
				int c = 0;
				long long psum = 0;
				rep(k,a[i].size()){
					while(a[j].size()!=c && a[j][c]<a[i][k]){
						psum += a[j][c];
						c++;
					}
					tsum += c * a[i][k] - psum;
				}
			}
			swap(i,j);
			{
				int c = 0;
				long long psum = 0;
				rep(k,a[i].size()){
					while(a[j].size()!=c && a[j][c]<a[i][k]){
						psum += a[j][c];
						c++;
					}
					tsum += c * a[i][k] - psum;
				}
			}
			swap(i,j);
			tsum *= abs(i-j);
			ans += tsum;
		}
	}
	ans *= 2;
	cout<<ans<<endl;
	return 0;
}
0