結果

問題 No.1788 Same Set
コンテスト
ユーザー tails
提出日時 2021-12-21 15:04:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 204 ms / 4,000 ms
コード長 7,140 bytes
コンパイル時間 2,390 ms
コンパイル使用メモリ 185,732 KB
実行使用メモリ 14,848 KB
最終ジャッジ日時 2024-09-15 15:35:18
合計ジャッジ時間 8,665 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC optimize("inline")
#include<bits/stdc++.h>
using namespace std;
void*wmem;
char memarr[96000000];
template<class T> inline void walloc1d(T **arr, int x, void **mem = &wmem){
  static int skip[16] = {0, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1};
  (*mem) = (void*)( ((char*)(*mem)) + skip[((unsigned long long)(*mem)) & 15] );
  (*arr)=(T*)(*mem);
  (*mem)=((*arr)+x);
}
template<class T> inline void walloc1d(T **arr, int x1, int x2, void **mem = &wmem){
  walloc1d(arr, x2-x1, mem);
  (*arr) -= x1;
}
inline int my_getchar_unlocked(){
  static char buf[1048576];
  static int s = 1048576;
  static int e = 1048576;
  if(s == e && e == 1048576){
    e = fread_unlocked(buf, 1, 1048576, stdin);
    s = 0;
  }
  if(s == e){
    return EOF;
  }
  return buf[s++];
}
inline void rd(int &x){
  int k;
  int m=0;
  x=0;
  for(;;){
    k = my_getchar_unlocked();
    if(k=='-'){
      m=1;
      break;
    }
    if('0'<=k&&k<='9'){
      x=k-'0';
      break;
    }
  }
  for(;;){
    k = my_getchar_unlocked();
    if(k<'0'||k>'9'){
      break;
    }
    x=x*10+k-'0';
  }
  if(m){
    x=-x;
  }
}
template<class T> struct segtree_ph{
  int N;
  int logN;
  T*val;
  void malloc(int maxN, int once = 0);
  void walloc(int maxN, int once = 0, void **mem = &wmem);
  void free(void);
  T& operator[](int i);
  void setN(int n, int zerofill = 1, int dobuild = 1);
  void build(void);
  inline void build(int a);
  inline void change(int a, T v);
  inline void changed(int a);
  inline T get(int a, int b);
}
;
struct segt{
  int c;
  int d;
  int e;
}
;
inline segt segtree_ph_func(segt a,segt b){
  segt r;
  int de=a.d-b.e;
  r.d=(de>=0?de:0)+b.d;
  r.e=(de>=0?0:-de)+a.e;
  r.c=(de>=0?a.c:0)+(de<=0?b.c:0);
  return r;
}
int pa[400000+1];
int pb[400000+1];
int main(){
  int i;
  wmem = memarr;
  int n;
  rd(n);
  int a[n];
  {
    int Nzj9Y0kE;
    for(Nzj9Y0kE=(0);Nzj9Y0kE<(n);Nzj9Y0kE++){
      rd(a[Nzj9Y0kE]);
    }
  }
  int b[n];
  {
    int QAAnbtfa;
    for(QAAnbtfa=(0);QAAnbtfa<(n);QAAnbtfa++){
      rd(b[QAAnbtfa]);
    }
  }
  segtree_ph<segt> t;
  t.walloc(n+3,1);
  t[0]=t[n+2]=segt{0,0,0};
  {
    int R5MtCiij;
    for(R5MtCiij=(1);R5MtCiij<((n+1) + 1);R5MtCiij++){
      t[R5MtCiij]=segt{1,0,0};
    }
  }
  t.build();
  long z=0;
  for(i=(0);i<(n);i++){
    int ai=a[i];
    int bi=b[i];
    if(pa[ai]<pb[ai]){
      {
        int j=pa[ai];
        t[j+0].d-=1;
        t.changed(j+0);
      }
      {
        int j=pb[ai];
        t[j+1].e-=1;
        t.changed(j+1);
      }
    }
    else if(pb[ai]<pa[ai]){
      {
        int j=pb[ai];
        t[j+0].d-=1;
        t.changed(j+0);
      }
      {
        int j=pa[ai];
        t[j+1].e-=1;
        t.changed(j+1);
      }
    }
    if(ai!=bi){
      if(pa[bi]<pb[bi]){
        {
          int j=pa[bi];
          t[j+0].d-=1;
          t.changed(j+0);
        }
        {
          int j=pb[bi];
          t[j+1].e-=1;
          t.changed(j+1);
        }
      }
      else if(pb[bi]<pa[bi]){
        {
          int j=pb[bi];
          t[j+0].d-=1;
          t.changed(j+0);
        }
        {
          int j=pa[bi];
          t[j+1].e-=1;
          t.changed(j+1);
        }
      }
      {
        int j=pb[ai];
        t[j+0].d+=1;
        t.changed(j+0);
      }
      {
        int j=pa[bi];
        t[j+0].d+=1;
        t.changed(j+0);
      }
      {
        int j=i+1;
        t[j+1].e+=2;
        t.changed(j+1);
      }
    }
    pa[ai]=i+1;
    pb[bi]=i+1;
    z+=t.val[1].c+i-n;
  }
  dprintf(1,"%ld\n",z);
  return 0;
}
template<class T> void segtree_ph<T>::malloc(int maxN, int once /*= 0*/){
  int i;
  for(i=1;i<maxN;i*=2){
    ;
  }
  val = new T[2*i];
  if(once){
    setN(maxN);
  }
}
template<class T> void segtree_ph<T>::walloc(int maxN, int once /*= 0*/, void **mem /*= &wmem*/){
  int i;
  for(i=1;i<maxN;i*=2){
    ;
  }
  walloc1d(&val, 2*i, mem);
  if(once){
    setN(maxN);
  }
}
template<class T> void segtree_ph<T>::free(void){
  delete [] val;
}
template<class T> T& segtree_ph<T>::operator[](int i){
  return val[N+i];
}
template<class T> void segtree_ph<T>::setN(int n, int zerofill /*= 1*/, int dobuild /*= 1*/){
  int i;
  for(i=1,logN=0;i<n;i*=2,logN++){
    ;
  }
  N = i;
  if(dobuild){
    build();
  }
}
template<class T> void segtree_ph<T>::build(void){
  for(int i=N-1;i;i--){
    val[i] = segtree_ph_func(val[2*i], val[2*i+1]);
  }
}
template<class T> inline void segtree_ph<T>::build(int a){
  while(a > 1){
    a /= 2;
    val[a] = segtree_ph_func(val[2*a], val[2*a+1]);
  }
}
template<class T> inline void segtree_ph<T>::change(int a, T v){
  val[a+N] = v;
  build(a+N);
}
template<class T> inline void segtree_ph<T>::changed(int a){
  build(a+N);
}
template<class T> inline T segtree_ph<T>::get(int a, int b){
  T res;
  T tmp;
  int fga = 0;
  int fgb = 0;
  a += N;
  b += N;
  while(a < b){
    if(a%2){
      if(fga){
        res = segtree_ph_func(res, val[a]);
      }
      else{
        res = val[a];
        fga = 1;
      }
      a++;
    }
    if(b%2){
      b--;
      if(fgb){
        tmp = segtree_ph_func(val[b], tmp);
      }
      else{
        tmp = val[b];
        fgb = 1;
      }
    }
    a /= 2;
    b /= 2;
  }
  if(fga==1 && fgb==0){
    return res;
  }
  if(fga==0 && fgb==1){
    return tmp;
  }
  if(fga==1 && fgb==1){
    return segtree_ph_func(res, tmp);
  }
  return res;
}
// modified
// cLay version 20211024-1

// --- original code ---
// struct segt{
// 	int c;
// 	int d,e;
// };
// 
// inline
// segt segtree_ph_func(segt a,segt b){
// 	segt r;
// 	int de=a.d-b.e;
// 	r.d=(de>=0?de:0)+b.d;
// 	r.e=(de>=0?0:-de)+a.e;
// 	r.c=(de>=0?a.c:0)+(de<=0?b.c:0);
// 	return r;
// }
// 
// int pa[4d5+1],pb[4d5+1];
// 
// {
// 	int@n,@a[n],@b[n];
// 	
// 	segtree_ph<segt> t;
// 	t.walloc(n+3,1);
// //	t[0..n+2]=segt{0,0,0};
// 	t[0]=t[n+2]=segt{0,0,0};
// 	t[1..n+1]=segt{1,0,0};
// 	t.build();
// 	long z=0;
// 	rep(i,n){
// 		int ai=a[i];
// 		int bi=b[i];
// 		if(pa[ai]<pb[ai]){
// 			{
// 				int j=pa[ai];
// 				t[j+0].d-=1;
// 				t.changed(j+0);
// 			}
// 			{
// 				int j=pb[ai];
// 				t[j+1].e-=1;
// 				t.changed(j+1);
// 			}
// 		}else if(pb[ai]<pa[ai]){
// 			{
// 				int j=pb[ai];
// 				t[j+0].d-=1;
// 				t.changed(j+0);
// 			}
// 			{
// 				int j=pa[ai];
// 				t[j+1].e-=1;
// 				t.changed(j+1);
// 			}
// 		}
// 		if(ai!=bi){
// 			if(pa[bi]<pb[bi]){
// 				{
// 					int j=pa[bi];
// 					t[j+0].d-=1;
// 					t.changed(j+0);
// 				}
// 				{
// 					int j=pb[bi];
// 					t[j+1].e-=1;
// 					t.changed(j+1);
// 				}
// 			}else if(pb[bi]<pa[bi]){
// 				{
// 					int j=pb[bi];
// 					t[j+0].d-=1;
// 					t.changed(j+0);
// 				}
// 				{
// 					int j=pa[bi];
// 					t[j+1].e-=1;
// 					t.changed(j+1);
// 				}
// 			}
// 			{
// 				int j=pb[ai];
// 				t[j+0].d+=1;
// 				t.changed(j+0);
// 			}
// 			{
// 				int j=pa[bi];
// 				t[j+0].d+=1;
// 				t.changed(j+0);
// 			}
// 			{
// 				int j=i+1;
// 				t[j+1].e+=2;
// 				t.changed(j+1);
// 			}
// 		}
// 		pa[ai]=i+1;
// 		pb[bi]=i+1;
// 		z+=t.val[1].c+i-n;
// 	}
// 	dprintf(1,"%ld\n",z);
// }
0