RUPC day1 C: 一致
問題ページ
Aizu Online Judge Arena
解法
sigmaさんのコードを参考にさせてもらった。
multi_setの同型判定をハッシュを用いて行う。a[i]に対応するハッシュ値を乱数で決める。multi_setのハッシュ値を (存在している値のハッシュ値) * (存在している数) の和とする。
[0,i]の区間のmulti_setのハッシュ値と[n-i-1,n-1]のmulti_setのハッシュ値を比較して一致していればそのインデックスを出力する。
ハッシュ値の衝突しなさそうな決め方がよくわかっていないのでもうちょっとちゃんと考えたい。
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; #define int ll using VI = vector<int>; using VVI = vector<VI>; using PII = pair<int, int>; #define FOR(i, a, n) for (ll i = (ll)a; i < (ll)n; ++i) #define REP(i, n) FOR(i, 0, n) #define ALL(x) x.begin(), x.end() #define PB push_back const ll LLINF = (1LL<<60); const int INF = (1LL<<30); const int MOD = 1000000007; template <typename T> T &chmin(T &a, const T &b) { return a = min(a, b); } template <typename T> T &chmax(T &a, const T &b) { return a = max(a, b); } template <typename T> bool IN(T a, T b, T x) { return a<=x&&x<b; } template<typename T> T ceil(T a, T b) { return a/b + !!(a%b); } template<class S,class T> ostream &operator <<(ostream& out,const pair<S,T>& a){ out<<'('<<a.first<<','<<a.second<<')'; return out; } template<class T> ostream &operator <<(ostream& out,const vector<T>& a){ out<<'['; REP(i, a.size()) {out<<a[i];if(i!=a.size()-1)out<<',';} out<<']'; return out; } int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0}; signed main(void) { cin.tie(0); ios::sync_with_stdio(false); int n; cin >> n; VI a(n); REP(i, n) cin >> a[i]; using D = uniform_int_distribution<ll>; random_device rd; mt19937 gen(rd()); map<int, int> hash; REP(i, n) { if(hash.find(a[i]) != hash.end()) continue; hash[a[i]] = D(0,LLINF)(gen); } int x = 0, y = 0; VI ans; REP(i, n) { x += hash[a[i]]; y += hash[a[n-i-1]]; if(x == y) ans.PB(i+1); } REP(i, ans.size()) { cout << ans[i] << (i==ans.size()-1?"\n":" "); } return 0; }