結果

問題 No.404 部分門松列
コンテスト
ユーザー rickytheta
提出日時 2016-08-05 21:20:25
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 937 ms / 2,000 ms
コード長 2,258 bytes
コンパイル時間 2,615 ms
コンパイル使用メモリ 175,096 KB
実行使用メモリ 30,848 KB
最終ジャッジ日時 2024-06-28 17:03:46
合計ジャッジ時間 12,278 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:56:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   56 |   scanf("%d",&n);
      |   ~~~~~^~~~~~~~~
main.cpp:57:16: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   57 |   REP(i,n)scanf("%d",a+i);
      |           ~~~~~^~~~~~~~~~
main.cpp:95:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   95 |   scanf("%d",&q);
      |   ~~~~~^~~~~~~~~
main.cpp:98:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   98 |     scanf("%d%d",&x,&y);
      |     ~~~~~^~~~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;

typedef int _loop_int;
#define REP(i,n) for(_loop_int i=0;i<(_loop_int)(n);++i)
#define FOR(i,a,b) for(_loop_int i=(_loop_int)(a);i<(_loop_int)(b);++i)
#define FORR(i,a,b) for(_loop_int i=(_loop_int)(b)-1;i>=(_loop_int)(a);--i)

#define DEBUG(x) cout<<#x<<": "<<x<<endl
#define DEBUG_VEC(v) cout<<#v<<":";REP(i,v.size())cout<<" "<<v[i];cout<<endl
#define ALL(a) (a).begin(),(a).end()

#define CHMIN(a,b) a=min((a),(b))
#define CHMAX(a,b) a=max((a),(b))

// mod
const ll MOD = 1000000007ll;
#define FIX(a) ((a)%MOD+MOD)%MOD

// floating
typedef double Real;
const Real EPS = 1e-11;
#define EQ0(x) (abs(x)<EPS)
#define EQ(a,b) (abs(a-b)<EPS)
typedef complex<Real> P;

struct Fenwick {
  int n;
  vl bit;
  Fenwick(int _n):n(_n+1){
    bit.assign(n+1,0);
  }
  void add(int a,ll w){
    for(int x=a;x<=n;x+=x&-x)
      bit[x] += w;
  }
  ll sum(int a){
    ll r=0;
    for(int x=a;x>0;x-=x&-x)
      r+=bit[x];
    return r;
  }
};

int n;
int a[252521];

int main(){
  scanf("%d",&n);
  REP(i,n)scanf("%d",a+i);
  // 座圧
  set<int> zat;
  REP(i,n)zat.insert(a[i]);
  zat.insert(1252583025);
  map<int,int> mp;
  map<int,int> revmp;
  int it = 1;
  for(int p:zat){
    mp[p] = it;
    revmp[it] = p;
    it++;
  }
  REP(i,n){
    a[i] = mp[a[i]];
  }

  Fenwick ans = Fenwick(n+10);
  Fenwick left = Fenwick(n+10);
  Fenwick mul = Fenwick(n+10);
  Fenwick right = Fenwick(n+10);
  REP(i,n)right.add(a[i],1);
  REP(i,n){
    right.add(a[i],-1);
    mul.add(a[i],-(left.sum(a[i])-left.sum(a[i]-1)));
    ll sm = left.sum(a[i]-1) * right.sum(a[i]-1);
    sm += (left.sum(n+5)-left.sum(a[i])) * (right.sum(n+5)-right.sum(a[i]));
    sm -= mul.sum(a[i]-1);
    sm -= (mul.sum(n+5)-mul.sum(a[i]));
    // DEBUG(sm);
    ans.add(a[i],sm);
    left.add(a[i],1);
    mul.add(a[i],right.sum(a[i])-right.sum(a[i]-1));
  }

  // REP(i,n+1)DEBUG(ans.sum(i));

  int q;
  scanf("%d",&q);
  while(q--){
    int x,y;
    scanf("%d%d",&x,&y);
    x = *zat.lower_bound(x); x = mp[x];
    y = *zat.upper_bound(y); y = mp[y]-1;
    ll p = ans.sum(y) - ans.sum(x-1);
    printf("%lld\n",p);
  }
  return 0;
}
0