結果

問題 No.805 UMG
コンテスト
ユーザー outline
提出日時 2020-03-27 16:38:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 62 ms / 2,000 ms
コード長 1,634 bytes
コンパイル時間 1,040 ms
コンパイル使用メモリ 117,904 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-18 13:59:20
合計ジャッジ時間 2,132 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <deque>
#include <numeric>
#include <bitset>
#include <iomanip>
#include <cassert>
#include <chrono>
#include <random>
#include <limits>
#include <iterator>
#include <functional>
#include <sstream>
#include <complex>
using namespace std;

typedef long long ll;
typedef uint64_t ull;
typedef pair<int, int> P;
typedef pair<int, double> Pid;
typedef pair<double, int> Pdi;
typedef pair<ll, int> Pl;
typedef pair<ll, ll> Pll;
typedef pair<int, pair<int, int>> PP;
typedef pair<P, int> PPi;
constexpr double PI = 3.1415926535897932;   // acos(-1)
constexpr double EPS = 1e-9;
constexpr int INF = 1001001001;
constexpr int mod = 1000000007;
// constexpr int mod = 998244353;

#define chmax(x, y) x = max(x, y)
#define chmin(x, y) x = min(x, y)
#define chadd(x, y) x = (x + y) % mod

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n;
    string s;
    cin >> n >> s;
    vector<vector<int>> pos(2);
    set<int> st;
    for(int i = 0; i < n; ++i){
        if(s[i] == 'U') pos[0].emplace_back(i);
        else if(s[i] == 'M')    pos[1].emplace_back(i);
        else    st.insert(i);
    }
    int sz0 = pos[0].size(), sz1 = pos[1].size();
    int ans = 0;
    for(int i = 0; i < sz0; ++i){
        for(int j = 0; j < sz1; ++j){
            if(pos[0][i] > pos[1][j])   continue;
            int dif = pos[1][j] - pos[0][i];
            if(st.find(pos[1][j] + dif) != st.end())    ++ans;
        }
    }
    cout << ans << endl;
}
0