#737482 (C++17) No.430 文字列検索

提出ソース

結果

問題	No.430 文字列検索
ユーザー	MtSaka
提出日時	2022-02-06 08:37:01
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	43 ms / 2,000 ms
コード長	6,661 bytes
コンパイル時間	1,947 ms
コンパイル使用メモリ	211,528 KB
実行使用メモリ	6,820 KB
最終ジャッジ日時	2024-11-10 00:59:20
合計ジャッジ時間	2,766 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,820 KB
testcase_01	AC	43 ms 6,816 KB
testcase_02	AC	13 ms 6,816 KB
testcase_03	AC	15 ms 6,816 KB
testcase_04	AC	2 ms 6,820 KB
testcase_05	AC	2 ms 6,816 KB
testcase_06	AC	1 ms 6,820 KB
testcase_07	AC	2 ms 6,820 KB
testcase_08	AC	13 ms 6,816 KB
testcase_09	AC	2 ms 6,816 KB
testcase_10	AC	3 ms 6,820 KB
testcase_11	AC	31 ms 6,820 KB
testcase_12	AC	29 ms 6,816 KB
testcase_13	AC	30 ms 6,816 KB
testcase_14	AC	25 ms 6,816 KB
testcase_15	AC	21 ms 6,816 KB
testcase_16	AC	13 ms 6,816 KB
testcase_17	AC	13 ms 6,816 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#line 1 "library/template/template.hpp"
//#pragma GCC target("avx")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#define overload4(a,b,c,d,e,...) e
#define overload3(a,b,c,d,...) d
#define rep1(a) for(ll i=0;i<(ll)(a);i++)
#define rep2(i,a) for(ll i=0;i<(ll)(a);i++)
#define rep3(i,a,b) for(ll i=(ll)(a);i<(ll)(b);i++)
#define rep4(i,a,b,c) for(ll i=(ll)(a);i<(ll)(b);i+=(ll)(c))
#define rep(...) overload4(__VA_ARGS__, rep4, rep3, rep2, rep1)(__VA_ARGS__)
#define rrep1(a) for(ll i=(ll)(a)-1;i>=0;i--)
#define rrep2(i,a) for(ll i=(ll)(a)-1;i>=0;i--)
#define rrep3(i,a,b) for(ll i=(ll)(b)-1;i>=(ll)(a);i--)
#define rrep(...) overload3(__VA_ARGS__,rrep3,rrep2,rrep1)(__VA_ARGS__)
#define all1(i) begin(i),end(i)
#define all2(i,a) begin(i),begin(i)+a
#define all3(i,a,b) begin(i)+a,begin(i)+b
#define all(...) overload3(__VA_ARGS__,all3,all2,all1)(__VA_ARGS__)
#define rall(n) (n).rbegin(),(n).rend()
#define INT(...) int __VA_ARGS__;scan(__VA_ARGS__)
#define LL(...) ll __VA_ARGS__;scan(__VA_ARGS__)
#define STR(...) string __VA_ARGS__;scan(__VA_ARGS__)
#define CHR(...) char __VA_ARGS__;scan(__VA_ARGS__)
#define DBL(...) double __VA_ARGS__;scan(__VA_ARGS__)
#define LD(...) ld __VA_ARGS__;scan(__VA_ARGS__)
#define pb push_back
#define eb emplace_back
#define END(...) {print(__VA_ARGS__);return;}
using namespace std;
using ll=long long;
using ull=unsigned long long;
using ld=long double;
using vl=vector<ll>;
using vi=vector<int>;
using vs=vector<string>;
using vc=vector<char>;
using vvl=vector<vl>;
using pi=pair<int,int>;
using pl=pair<ll,ll>;
using vvc=vector<vc>;
using vd=vector<double>;
using vp=vector<pl>;
using vb=vector<bool>;
const int dx[8]={1,0,-1,0,1,-1,-1,1};
const int dy[8]={0,1,0,-1,1,1,-1,-1};
const ll MOD=1000000007;
const ll mod=998244353;
const ld EPS=1e-8;
const ld PI=3.1415926535897932384626;
template<typename T,typename U>
ostream &operator<<(ostream&os,const pair<T,U>&p){os<<p.first<<" "<<p.second;return os;}
template<typename T,typename U>
istream &operator>>(istream&is,pair<T,U>&p){is>>p.first>>p.second;return is;}
template<typename T>
ostream &operator<<(ostream&os,const vector<T>&v){for(auto it=v.begin();it!=v.end();){os<<*it<<((++it)!=v.end()?" ":"");}return os;}
template<typename T>
istream &operator>>(istream&is,vector<T>&v){for(T &in:v){is>>in;}return is;}
void scan(){}
template<class Head,class... Tail>
void scan(Head&head,Tail&... tail){cin>>head;scan(tail...);}
template<class T>
void print(const T &t){cout<<t<<'\n';}
template<class Head, class... Tail>
void print(const Head &head, const Tail &... tail){cout<<head<<' ';print(tail...);}
template<class... T>
void fin(const T &... a){print(a...);exit(0);}
template<typename T,typename U>
inline bool chmax(T&a,U b){return a<b&&(a=b,true);}
template<typename T,typename U>
inline bool chmin(T&a,U b){return a>b&&(a=b,true);}
template<typename T>
class infinity{
  public:
  static const T MAX=numeric_limits<T>::max();
  static const T MIN=numeric_limits<T>::min();
  static const T value=numeric_limits<T>::max()/2;
  static const T mvalue=numeric_limits<T>::min()/2;
};
#if __cplusplus <= 201402L
template<class T>const T infinity<T>::value;
template<class T>const T infinity<T>::mvalue;
template<class T>const T infinity<T>::MAX;
template<class T>const T infinity<T>::MIN;
#endif
template<typename T>const T inf=infinity<T>::value;
inline int popcnt(ull x){
#if __cplusplus>=202002L
return popcount(x);
#endif
x=(x&0x5555555555555555)+((x>>1)&0x5555555555555555);x=(x&0x3333333333333333)+((x>>2)&0x3333333333333333);x=(x&0x0f0f0f0f0f0f0f0f)+((x>>4)&0x0f0f0f0f0f0f0f0f);x=(x&0x00ff00ff00ff00ff)+((x>>8)&0x00ff00ff00ff00ff);x=(x&0x0000ffff0000ffff)+((x>>16)&0x0000ffff0000ffff);return (x&0x00000000ffffffff)+((x>>32)&0x00000000ffffffff);
}
void dump(const char&t){cerr<<t;}
void dump(const string&t){cerr<<t;}
void dump(const bool&t){cerr<<(t?"true":"false");}
template<typename T>
void dump(const T&t,enable_if_t<is_integral<T>::value>* =nullptr){string tmp;if(t==infinity<T>::value||t==infinity<T>::MAX)tmp="inf";if(t==infinity<T>::mvalue||t==infinity<T>::MIN)tmp="-inf";if(tmp.empty())tmp=to_string(t);cerr<<tmp;}
template <typename T>
void dump(const T&t,enable_if_t<!is_void<typename T::iterator>::value>* =nullptr){cerr<<"{";for(auto it=t.begin();it!=t.end();){dump(*it);cerr<<(++it==t.end()?"":",");}cerr<<"}";}
template<typename T,typename U>
void dump(const pair<T,U>&t){cerr<<"(";dump(t.first);cerr<<",";dump(t.second);cerr<<")";}
void trace(){cerr<<endl;}
template<typename Head,typename... Tail>
void trace(Head&&head,Tail&&... tail){dump(head);if(sizeof...(tail))cerr<<",";trace(forward<Tail>(tail)...);}
#ifdef ONLINE_JUDGE
#define debug(...) (void(0))
#else
#define debug(...) do{cerr<<#__VA_ARGS__<<"=";trace(__VA_ARGS__);}while(0)
#endif
struct IOSetup{IOSetup(){cin.tie(nullptr);ios::sync_with_stdio(false);cout.tie(0);cout<<fixed<<setprecision(12);cerr<<fixed<<setprecision(12);}};
/**
 * @brief Template(テンプレート)
*/
#line 1 "library/String/rolling_hash.hpp"
struct RollingHash{
  using ull=unsigned long long;
  using i128=__uint128_t;
  private:
  static const ull MOD=(1ull<<61)-1;
  static const ull MASK31=(1ull<<31)-1;
  static ull calc_mod(ull a){
    ull ret=(a&MOD)+(a>>61);
    if(ret>=MOD)ret-=MOD;
    return ret;
  }
  static ull calc_mul(ull a,ull b){
    i128 c=(i128)a*b;
    return calc_add(c&MOD,c>>61);
  }
  static ull calc_add(ull a,ull b){
    ull ret=a+b;
    if(ret>=MOD)ret-=MOD;
    return ret;
  }
  ull BASE;
  void init(){
    BASE=(1ull<<31)+(random_device()()&MASK31);
  }
  public:
  struct Hash{
    private:
    int n;
    ull BASE;
    vector<ull>hash,power;
    public:
    template<typename T>
    Hash(ull base,const T&s):BASE(base){
      n=s.size();
      hash.resize(n+1);
      for(int i=0;i<n;i++)hash[i+1]=calc_add(s[i],calc_mul(BASE,hash[i]));
      power.resize(n+1);power[0]=1;
      for(int i=0;i<n;i++)power[i+1]=calc_mul(power[i],BASE);
    }
    ull get_hash(int l,int r)const{
      return calc_add(hash[r],MOD-calc_mul(hash[l],power[r-l]));
    }
    ull get_all()const{
      return hash[n];
    }
  };
  RollingHash(){init();}
  template<typename T>
  Hash get_hash(const T&s)const{return Hash(BASE,s);}
  ull get_base()const{return BASE;}
};
/**
 * @brief Rolling Hash(ローリングハッシュ)
*/
#line 3 "code.cpp"
int main(){
  STR(s);
  RollingHash rh;
  auto h=rh.get_hash(s);
  LL(m);
  ll ans=0;
  set<ull>t;
  rep(i,m){
    STR(c);
    ull a=rh.get_hash(c).get_all();
    t.insert(a);
  }
  rep(i,1,11){
    rep(j,s.size()-i+1){
      ull a=h.get_hash(j,j+i);
      if(t.count(a))ans++;
    }
  }
  fin(ans);
}

yukicoder

結果

テストケース

ソースコード