結果

問題 No.1712 Read and Pile
ユーザー chineristACchineristAC
提出日時 2021-05-21 20:04:25
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 140 ms / 2,000 ms
コード長 3,034 bytes
コンパイル時間 9,748 ms
コンパイル使用メモリ 312,276 KB
実行使用メモリ 15,368 KB
最終ジャッジ日時 2024-09-17 16:55:10
合計ジャッジ時間 14,134 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 3 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 96 ms
11,276 KB
testcase_09 AC 130 ms
14,152 KB
testcase_10 AC 87 ms
10,880 KB
testcase_11 AC 99 ms
11,824 KB
testcase_12 AC 70 ms
10,556 KB
testcase_13 AC 89 ms
10,704 KB
testcase_14 AC 77 ms
10,516 KB
testcase_15 AC 110 ms
11,624 KB
testcase_16 AC 98 ms
11,484 KB
testcase_17 AC 92 ms
11,204 KB
testcase_18 AC 101 ms
11,020 KB
testcase_19 AC 89 ms
10,436 KB
testcase_20 AC 88 ms
10,792 KB
testcase_21 AC 105 ms
11,636 KB
testcase_22 AC 94 ms
11,652 KB
testcase_23 AC 122 ms
14,732 KB
testcase_24 AC 107 ms
14,472 KB
testcase_25 AC 126 ms
14,728 KB
testcase_26 AC 140 ms
15,240 KB
testcase_27 AC 135 ms
15,368 KB
testcase_28 AC 70 ms
9,968 KB
testcase_29 AC 106 ms
13,372 KB
testcase_30 AC 82 ms
11,504 KB
testcase_31 AC 38 ms
12,624 KB
testcase_32 AC 30 ms
10,520 KB
testcase_33 AC 61 ms
12,864 KB
testcase_34 AC 57 ms
12,220 KB
testcase_35 AC 82 ms
13,588 KB
testcase_36 AC 90 ms
14,752 KB
testcase_37 AC 85 ms
13,640 KB
testcase_38 AC 2 ms
5,376 KB
testcase_39 AC 14 ms
5,376 KB
testcase_40 AC 18 ms
6,308 KB
testcase_41 AC 19 ms
6,312 KB
testcase_42 AC 23 ms
6,436 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <queue>
#include <algorithm>
#include <cmath>
#include <iomanip>
#include <random>
#include <stdio.h>
#include <fstream>
#include <functional>
#include <atcoder/all>
#include "testlib.h"
using namespace std;
using namespace atcoder;

#define rep(i,n,c) for (int i=0;i<n;i+=c)
#define append push_back
#define all(x) (x).begin(), (x).end()

template<class T>
using vec = vector<T>;
template<class T>
using vvec = vec<vec<T>>;
template<class T>
using vvvec = vec<vvec<T>>;
using ll = long long;
using pii = pair<int,int>;
using pll = pair<ll,ll>;


template<class T>
bool chmin(T &a, T b){
  if (a>b){
    a = b;
    return true;
  }
  return false;
}

template<class T>
bool chmax(T &a, T b){
  if (a<b){
    a = b;
    return true;
  }
  return false;
}

template<class T>
T sum(vec<T> x){
  T res=0;
  for (auto e:x){
    res += e;
  }
  return res;
}

template<class T>
void printv(vec<T> x){
  for (auto e:x){
    cout<<e<<" ";
  }
  cout<<"\n";
}


const ll INF = 1e17;

const int MIN_N = 1;
const int MAX_N = 200000;
const int MIN_M = 1;
const int MAX_M = 200000;


using mint = modint998244353;
const mint i2 = ((mint)(2)).inv();

int main(int argc, char* argv[]){
  registerValidation(argc,argv);
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int N = inf.readInt(MIN_N,MAX_N,"N");
  inf.readSpace();
  int M = inf.readInt(MIN_M,MAX_M,"M");
  inf.readEoln();

  vec<int> A(N+M+1);
  for (int i=1;i<N+1;i++){
    A[i] = N-i+1;
  }
  A[0] = -2;
  for (int i=N+1;i<N+M+1;i++){
    A[i] = inf.readInt(-1,N);
    assert (A[i]!=0);
    if (i!=N+M){
      inf.readSpace();
    }
    else{
      inf.readEoln();
    }
  }
  inf.readEof();

  if (N==1){
    cout<<M<<endl;
    return 0;
  }
  else if (N==2){
    int pre=1;
    mint res=M;
    for (int i=N+1;i<N+M+1;i++){
      if (A[i]==-1){
        res += i2;
      }
      else{
        if (pre!=-1 and A[i]!=pre){
          res += 1;
        }
        else if (pre==-1){
          res += i2;
        }
      }
      pre = A[i];
    }
    cout<<res.val()<<endl;
    return 0;
  }
  
  mint p = (N-2) * ((mint)(N)).inv();
  mint ip = p.inv();

  vec<int> zero(N+M+1);
  rep(i,N+M+1,1){
    if (A[i]==-1){
      zero[i] = 1;
    }
    if (i){
      zero[i] += zero[i-1];
    }
  }

  vec<int> last(N+1);
  rep(i,N+1,1){
    last[i] = N-i+1;
  }

  fenwick_tree<mint> fwp(N+M+1);
  fenwick_tree<int> fwcnt(N+M+1);
  for (int i=1;i<N+1;i++){
    fwcnt.add(i,1);
    fwp.add(i,ip.pow(zero[i]));
  }

  mint res;
  int pre,k;
  for (int i=N+1;i<N+M+1;i++){
    if (A[i]==-1){
      res += N-1;
    }
    else{
      pre = last[A[i]];
      k = fwcnt.sum(1,pre);

      res += N-1;
      res += (fwp.sum(pre+1,i+1)) * p.pow(zero[i]);
      res -= k * p.pow(zero[i]-zero[pre]);

      fwcnt.add(pre,-1);
      fwp.add(pre,-ip.pow(zero[pre]));

      last[A[i]] = i;
      fwcnt.add(i,1);
      fwp.add(i,ip.pow(zero[i]));
    }
  }

  res /= 2;
  res += M;
  cout<<res.val()<<endl;
}
0