#463475 (C#(csc)) No.430 文字列検索

• 問題: No.430 文字列検索
• ユーザー: mnzn
• 提出日時: 2020-04-13 11:55:06
• 言語: C#(csc) (csc 3.9.0)
• 結果: AC
• 実行時間: 113 ms / 2,000 ms
• コード長: 7,134 bytes
• コンパイル時間: 834 ms
• コンパイル使用メモリ: 115,008 KB
• 実行使用メモリ: 43,844 KB
• 最終ジャッジ日時: 2024-11-10 00:42:26
• 合計ジャッジ時間: 2,430 ms
• ジャッジサーバーID （参考情報）: judge3 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	25 ms; 18,304 KB
testcase_01	AC	113 ms; 43,844 KB
testcase_02	AC	63 ms; 20,864 KB
testcase_03	AC	63 ms; 20,736 KB
testcase_04	AC	25 ms; 18,176 KB
testcase_05	AC	26 ms; 18,176 KB
testcase_06	AC	25 ms; 18,304 KB
testcase_07	AC	25 ms; 18,176 KB
testcase_08	AC	107 ms; 42,184 KB
testcase_09	AC	26 ms; 18,304 KB
testcase_10	AC	33 ms; 20,608 KB
testcase_11	AC	72 ms; 23,936 KB
testcase_12	AC	72 ms; 23,936 KB
testcase_13	AC	73 ms; 24,064 KB
testcase_14	AC	69 ms; 23,552 KB
testcase_15	AC	70 ms; 22,528 KB
testcase_16	AC	64 ms; 20,736 KB
testcase_17	AC	63 ms; 20,992 KB

コンパイルメッセージ

Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;
using System.Linq;

class Program
{
    static void Main(string[] args)
    {
        var S = sc.ReadStr();
        var M = sc.ReadInt();
        var rhs = new RollingHash(S);
        var ans = 0L;
        var map = new HashMap<long, long>();
        for (int i = 1; i <= 10; i++)
        {
            for (int j = 0; j <= S.Length - i; j++)
            {
                var h = rhs.Get(j, j + i);
                map[h]++;
            }
        }
        for (int i = 0; i < M; i++)
        {
            var T = sc.ReadStr();
            if (T.Length > S.Length) continue;
            var rht = new RollingHash(T);
            var hashT = rht.Get(0, T.Length);
            ans += map[hashT];
        }
        Console.WriteLine(ans);
    }
    class HashMap<K, V> : Dictionary<K, V> where V : new()
    {
        private V DefaltValue = new V();
        public HashMap() { }
        public HashMap(V defaultValue)
        {
            this.DefaltValue = new V();
        }
        new public V this[K i]
        {
            get
            {
                V v;
                return TryGetValue(i, out v) ? v : base[i] = this.DefaltValue;
            }
            set { base[i] = value; }
        }
    }


    class RollingHash
    {
        public const long B = (long)1e5 + 7;
        public const long MOD = (long)1e9 + 7;
        public string S { get; set; }
        public int N { get; set; }
        public long[] Power { get; set; }
        public long[] Hash { get; set; }
        public RollingHash(string s)
        {
            this.S = s;
            this.N = s.Length;
            // B^nを計算しておく
            this.Power = new long[this.N + 1];
            this.Power[0] = 1;
            for (int i = 0; i < N; i++)
            {
                this.Power[i + 1] = (this.Power[i] * B) % MOD;
            }
            // ハッシュを前計算する
            this.Hash = new long[this.N + 1];
            for (int i = 0; i < N; i++)
            {
                this.Hash[i + 1] = (this.Hash[i] * B + S[i]) % MOD;
            }
        }
        // [l, r) のハッシュ値を求める
        public long Get(int l, int r)
        {
            var hash = this.Hash[r] - (this.Hash[l] * this.Power[r - l] % MOD);
            if (hash < 0) hash += MOD;
            return hash;
        }
        // S[a:] and S[b:]
        public int GetLCP(int a, int b)
        {
            var len = Math.Min(N - a, N - b) + 1;
            var l = 0;
            var r = len;
            while (r - l > 1)
            {
                var mid = l + (r - l) / 2;
                var hashA = this.Get(a, a + mid);
                var hashB = this.Get(b, b + mid);
                if (hashA == hashB) l = mid;
                else r = mid;
            }
            return l;
        }
        // S[a:] and T[b:]
        public int GetLCP(RollingHash rollingHash, int a, int b)
        {
            var len = Math.Min(rollingHash.S.Length - a, N - b) + 1;
            var l = 0;
            var r = len;
            while (r - l > 1)
            {
                var mid = l + (r - l) / 2;
                var hash1 = this.Get(b, b + mid);
                var hash2 = rollingHash.Get(a, a + mid);
                if (hash1 == hash2) l = mid;
                else r = mid;
            }
            return l;
        }
    }

    static Scanner sc = new Scanner();
}
class Scanner
{
    string[] S = new string[0];
    int Index = 0;
    char[] Separators = new char[] { ' ' };

    public string Next()
    {
        if (this.Index < this.S.Length) return this.S[this.Index++];
        var line = "";
        while (line == "") line = Console.ReadLine();
        this.S = line.Split(this.Separators, StringSplitOptions.RemoveEmptyEntries);
        if (this.S.Length == 0) return this.Next();
        this.Index = 0;
        return this.S[this.Index++];
    }
    public string ReadStr()
    {
        return this.Next();
    }
    public char ReadChar()
    {
        return this.Next()[0];
    }
    public int ReadInt()
    {
        return int.Parse(this.Next());
    }
    public uint ReadUInt()
    {
        return uint.Parse(this.Next());
    }
    public long ReadLong()
    {
        return long.Parse(this.Next());
    }
    public double ReadDouble()
    {
        return double.Parse(this.Next());
    }
    public Tuple<int, int> ReadTup(int add = 0)
    {
        return Tuple.Create(this.ReadInt() + add, this.ReadInt() + add);
    }
    public Tuple<long, long> ReadTupLong(int add = 0)
    {
        return Tuple.Create(this.ReadLong() + add, this.ReadLong() + add);
    }
    public Tuple<int, int, int> ReadTup3(int add = 0)
    {
        return Tuple.Create(this.ReadInt() + add, this.ReadInt() + add, this.ReadInt() + add);
    }
    public Tuple<int, int, int, int> ReadTup4(int add = 0)
    {
        return Tuple.Create(this.ReadInt() + add, this.ReadInt() + add, this.ReadInt() + add, this.ReadInt() + add);
    }
    public int[] ReadIntArray(int n)
    {
        var array = new int[n];
        for (int i = 0; i < array.Length; i++)
        {
            array[i] = this.ReadInt();
        }
        return array;
    }
    public long[] ReadLongArray(int n)
    {
        var array = new long[n];
        for (int i = 0; i < array.Length; i++)
        {
            array[i] = this.ReadLong();
        }
        return array;
    }
    public double[] ReadDoubleArray(int n)
    {
        var array = new double[n];
        for (int i = 0; i < array.Length; i++)
        {
            array[i] = this.ReadDouble();
        }
        return array;
    }
    public char[] ReadCharArray(int n)
    {
        var array = new char[n];
        for (int i = 0; i < array.Length; i++)
        {
            array[i] = this.ReadChar();
        }
        return array;
    }
    public string[] ReadStrArray(int n)
    {
        var array = new string[n];
        for (int i = 0; i < array.Length; i++)
        {
            array[i] = this.ReadStr();
        }
        return array;
    }
    public Tuple<long, long>[] ReadTupLongArray(int n, int add = 0)
    {
        var array = new Tuple<long, long>[n];
        for (int i = 0; i < n; i++)
        {
            array[i] = this.ReadTupLong(add);
        }
        return array;
    }
    public Tuple<int, int>[] ReadTupArray(int n, int add = 0)
    {
        var array = new Tuple<int, int>[n];
        for (int i = 0; i < n; i++)
        {
            array[i] = this.ReadTup(add);
        }
        return array;
    }
    public Tuple<int, int, int>[] ReadTup3Array(int n, int add = 0)
    {
        var array = new Tuple<int, int, int>[n];
        for (int i = 0; i < n; i++)
        {
            array[i] = this.ReadTup3(add);
        }
        return array;
    }
    public Tuple<int, int, int, int>[] ReadTup4Array(int n, int add = 0)
    {
        var array = new Tuple<int, int, int, int>[n];
        for (int i = 0; i < n; i++)
        {
            array[i] = this.ReadTup4(add);
        }
        return array;
    }
}