結果

問題 No.1788 Same Set
ユーザー tute7627
提出日時 2021-12-01 20:41:27
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 1,034 bytes
コンパイル時間 11,044 ms
コンパイル使用メモリ 278,652 KB
最終ジャッジ日時 2025-01-26 02:41:52
ジャッジサーバーID
（参考情報） 		judge2 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 4 RE * 13 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

#include<bits/stdc++.h>
using namespace std;

int main(){
  int n;
  cin >> n;
  vector<int>a(n), b(n);
  for(int i = 0; i < n; i++)cin >> a[i];
  for(int i = 0; i < n; i++)cin >> b[i];
  vector<int>c(n);
  const int sz = 200005;
  vector<int> pre_a(sz, -1), pre_b(sz, -1);
  long long ans = 0;
  int csum = 0;
  for(int i = 0; i < n; i++){
    vector<int>upd;
    upd.push_back(a[i]);
    if(a[i] != b[i])upd.push_back(b[i]);
    for(auto val:upd){
      int l = min(pre_a[val], pre_b[val]);
      int r = max(pre_a[val], pre_b[val]);
      for(int i = l + 1; i < r + 1; i++){
        c[i]--;
        if(c[i] == 0)csum--;
      }
    }
    pre_a[a[i]] = i;
    pre_b[b[i]] = i;
    for(auto val:upd){
      int l = min(pre_a[val], pre_b[val]);
      int r = max(pre_a[val], pre_b[val]);
      for(int i = l + 1; i < r + 1; i++){
        if(c[i] == 0)csum++;
        c[i]++;
      }
    }
    ans += i + 1 - csum;
  }
  cout << ans << endl;
}
0