結果

問題 No.1996 <><
コンテスト
ユーザー umezo
提出日時 2022-07-07 07:21:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 539 ms / 2,000 ms
コード長 1,510 bytes
コンパイル時間 2,361 ms
コンパイル使用メモリ 205,380 KB
最終ジャッジ日時 2025-01-30 05:08:43
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;

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

const int MOD=1e9+7;

template<typename T>
struct BIT{
private:
  vector<T> A;
  const int n;
  
public:
  BIT(int _n) : A(_n+1,0), n(_n){}
  
  T sum(int i){
    T s=0;
    while(i>0){
      s=(s+A[i])%MOD;
      i-=i&-i;
    }
    return s;
  }
  
  T sum(int i,int j){
    return (sum(j)-sum(i-1)+MOD)%MOD;
  }
  
  void add(int i,T x){
    while(i<=n){
      A[i]=(A[i]+x)%MOD;
      i+=i&-i;
    }
  }

  void update(int i,T x){
    T tmp=sum(i,i);
    if(tmp!=x) add(i,(x-tmp+MOD)%MOD);
  }
};

template <typename T>
vector<T> compress(vector<T> &X){
  vector<T> vals=X;
  sort(vals.begin(),vals.end());
  vals.erase(unique(vals.begin(), vals.end()),vals.end());
  for(int i=0;i<(int)X.size(); i++){
    X[i]=lower_bound(vals.begin(),vals.end(),X[i])-vals.begin();
  }
  return vals;
}

ll dp[1515][1515];

int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int n,k;
  cin>>n>>k;
  vector<ll> A(n+1);
  string s;
  rep(i,n) cin>>A[i+1];
  cin>>s;
  
  compress(A);
  vector<BIT<ll>> B(k+1,BIT<ll>(1515));
  
  for(int i=1;i<=n;i++){
    for(int j=1;j<=k;j++){
      if(s[j-1]=='<') dp[i][j]=B[j-1].sum(1,A[i]-1);
      else dp[i][j]=B[j-1].sum(A[i]+1,n);
      B[j].add(A[i],dp[i][j]);
    }
    dp[i][0]=1;
    B[0].add(A[i],dp[i][0]);
  }
  ll ans=0;
  for(int i=k+1;i<=n;i++) ans=(ans+dp[i][k])%MOD;
  cout<<ans<<endl;

  return 0;
}
0