結果

問題 No.2419 MMA文字列2
ユーザー ragnaragna
提出日時 2023-08-12 14:05:15
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 173 ms / 2,000 ms
コード長 2,808 bytes
コンパイル時間 1,777 ms
コンパイル使用メモリ 175,588 KB
実行使用メモリ 155,156 KB
最終ジャッジ日時 2024-04-30 04:57:12
合計ジャッジ時間 4,644 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 1 ms
6,944 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 1 ms
6,944 KB
testcase_05 AC 1 ms
6,940 KB
testcase_06 AC 1 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 1 ms
6,940 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 1 ms
6,940 KB
testcase_12 AC 39 ms
37,248 KB
testcase_13 AC 7 ms
8,696 KB
testcase_14 AC 79 ms
68,480 KB
testcase_15 AC 45 ms
43,392 KB
testcase_16 AC 83 ms
71,552 KB
testcase_17 AC 23 ms
23,680 KB
testcase_18 AC 92 ms
74,664 KB
testcase_19 AC 93 ms
76,672 KB
testcase_20 AC 26 ms
26,368 KB
testcase_21 AC 17 ms
19,328 KB
testcase_22 AC 169 ms
154,932 KB
testcase_23 AC 170 ms
154,968 KB
testcase_24 AC 169 ms
154,916 KB
testcase_25 AC 173 ms
154,964 KB
testcase_26 AC 20 ms
21,376 KB
testcase_27 AC 167 ms
154,980 KB
testcase_28 AC 172 ms
155,156 KB
testcase_29 AC 172 ms
154,980 KB
testcase_30 AC 168 ms
154,932 KB
testcase_31 AC 168 ms
154,956 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>

/// @brief Union-Find 木
/// @note 1.4 高速化 + 省メモリ化
/// @see https://zenn.dev/reputeless/books/standard-cpp-for-competitive-programming/viewer/union-find
class UnionFind
{
public:

	UnionFind() = default;

	/// @brief Union-Find 木を構築します。
	/// @param n 要素数
	explicit UnionFind(size_t n)
		: m_parentsOrSize(n, -1) {}

	/// @brief 頂点 i の root のインデックスを返します。
	/// @param i 調べる頂点のインデックス
	/// @return 頂点 i の root のインデックス
	int find(int i)
	{
		if (m_parentsOrSize[i] < 0)
		{
			return i;
		}

		// 経路圧縮
		return (m_parentsOrSize[i] = find(m_parentsOrSize[i]));
	}

	/// @brief a のグループと b のグループを統合します。
	/// @param a 一方のインデックス
	/// @param b 他方のインデックス
	void merge(int a, int b)
	{
		a = find(a);
		b = find(b);

		if (a != b)
		{
			// union by size (小さいほうが子になる)
			if (-m_parentsOrSize[a] < -m_parentsOrSize[b])
			{
				std::swap(a, b);
			}

			m_parentsOrSize[a] += m_parentsOrSize[b];
			m_parentsOrSize[b] = a;
		}
	}

	/// @brief a と b が同じグループに属すかを返します。
	/// @param a 一方のインデックス
	/// @param b 他方のインデックス
	/// @return a と b が同じグループに属す場合 true, それ以外の場合は false
	bool connected(int a, int b)
	{
		return (find(a) == find(b));
	}

	/// @brief i が属するグループの要素数を返します。
	/// @param i インデックス
	/// @return i が属するグループの要素数
	int size(int i)
	{
		return -m_parentsOrSize[find(i)];
	}

private:

	// m_parentsOrSize[i] は i の 親,
	// ただし root の場合は (-1 * そのグループに属する要素数)
	std::vector<int> m_parentsOrSize;
};
using namespace std;
typedef long long ll;
#define rep(i,a,b) for(ll i=(ll)(a);i<(ll)(b);i++)
#define rrep(i,a,b) for(ll i=(ll)(a-1);i>=(ll)(b);i--)
#define MOD 998244353
//#define MOD 1e9+7
#define INF 1e16
//#define PI numbers::pi
//#define E numbers::e
template <typename T> bool chmax(T& a,T b){if(a<b){a=b;return 1;}return 0;}
template <typename T> bool chmin(T& a,T b){if(a>b){a=b;return 1;}return 0;}

int main(){
    string s; cin >> s;
    vector<vector<vector<ll>>> dp(s.size()+1,vector<vector<ll>>(3,vector<ll>(26,0)));
    rep(i,0,s.size()){
        ll a=s[i]-'A';
        rep(j,0,26){
            dp[i+1][0][j]=dp[i][0][j];
            dp[i+1][1][j]=dp[i][1][j];
            dp[i+1][2][j]=dp[i][2][j];
        }
        dp[i+1][0][a]++;
        dp[i+1][1][a]+=dp[i][0][a];
        rep(j,0,26){
            if(j!=a) dp[i+1][2][a]+=dp[i][1][j];
        }
    }
    ll ans=0;
    rep(i,0,26) ans+=dp[s.size()][2][i];
    cout << ans << endl;
}
0