結果

問題 No.430 文字列検索
ユーザー yuimyoyuimyo
提出日時 2023-12-09 12:13:13
言語 C#
(.NET 8.0.203)
結果
AC  
実行時間 88 ms / 2,000 ms
コード長 11,467 bytes
コンパイル時間 6,822 ms
コンパイル使用メモリ 166,488 KB
実行使用メモリ 189,852 KB
最終ジャッジ日時 2024-11-10 01:09:32
合計ジャッジ時間 8,700 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 45 ms
28,152 KB
testcase_01 AC 88 ms
30,192 KB
testcase_02 AC 80 ms
30,456 KB
testcase_03 AC 79 ms
30,332 KB
testcase_04 AC 47 ms
28,032 KB
testcase_05 AC 45 ms
27,904 KB
testcase_06 AC 45 ms
27,904 KB
testcase_07 AC 44 ms
27,900 KB
testcase_08 AC 80 ms
29,432 KB
testcase_09 AC 46 ms
28,284 KB
testcase_10 AC 55 ms
29,056 KB
testcase_11 AC 80 ms
30,336 KB
testcase_12 AC 84 ms
30,336 KB
testcase_13 AC 81 ms
30,308 KB
testcase_14 AC 81 ms
30,080 KB
testcase_15 AC 87 ms
30,584 KB
testcase_16 AC 81 ms
30,208 KB
testcase_17 AC 79 ms
189,852 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (82 ms)。
MSBuild のバージョン 17.9.6+a4ecab324 (.NET)
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff # 

using Kzrnm.Competitive.IO;
using System;
using System.Buffers.Text;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.Text;
namespace Roliha.Problems
{
    //using ModInt = StaticModInt<Mod998244353>;
    internal partial class ProblemA
    {
        //public class RollingHash
        //{
        //    ulong mod = (1UL << 61) - 1;
        //    ulong mask30 = (1UL << 30) - 1;
        //    ulong mask31 = (1UL << 31) - 1;
        //    ulong mask61 = mod;
        //    private ulong mul(ulong a, ulong b)
        //    {
        //        ulong an = a >> 31;
        //        ulong ad = a & mask31;
        //        ulong bn = b >> 31;
        //        ulong bd = b & mask31;
        //        ulong mid = ad * bn + an * bd;
        //        ulong midn = mid >> 30;
        //        ulong midd = mid & mask30;
        //        return an * bn * 2 + midn + (midd << 31) + ad * bd;
        //    }

        //    private ulong calcMod(ulong x)
        //    {
        //        ulong xn = x >> 61;
        //        ulong xd = x & mask61;
        //        ulong res = xn + xd;
        //        if (res >= mod) res -= mod;
        //        return res;
        //    }


        //}

        static void Solve()
        {
            var S = cr.String();
            var M = cr.Int();
            var C = cr.Repeat<string>(M);
            var mod = (1UL << 61) - 1;
            var sizeMax = Math.Min(S.Length, 10);
            var b = new ulong[sizeMax + 1];
            b[0] = 1;
            b[1] = 27;
            for (int i = 1; i < sizeMax; i++)
                b[i + 1] = b[i] * b[1];

            var actualHash = new ulong[sizeMax + 1];
            for (int i = 0; i < sizeMax; i++)
                actualHash[i + 1] = (actualHash[i] * b[1] + (ulong)(S[i] - 'A' + 1)) % mod;

            var hashCnt = new Dictionary<ulong, int>();
            for(int i = 0; i < M; i++)
            {
                ulong expectedHash = 0;
                for (int j = 0; j < C[i].Length; j++)
                    expectedHash = (expectedHash * b[1] + (ulong)(C[i][j] - 'A' + 1)) % mod;
                if (!hashCnt.ContainsKey(expectedHash))
                    hashCnt[expectedHash] = 0;
                hashCnt[expectedHash]++;
            }

            var ans = 0;
            for(int i = 0; i < S.Length; i++)
            {
                for (int j = 1; j <= sizeMax; j++)
                {
                    if (hashCnt.ContainsKey(actualHash[j]))
                    {
                        //cw.WriteLine($"({i}, {j}) +{hashCnt[actualHash[j]]}");
                        ans += hashCnt[actualHash[j]];
                    }
                    if (i + j >= S.Length) break;
                    actualHash[j] = (actualHash[j] * b[1] + (ulong)(S[i + j] - 'A' + 1) + mod - (ulong)(S[i] - 'A' + 1) * b[j]) % mod; 
                }
            }

            cw.WriteLine(ans);
        }
    }
}

#region Dependency resolution
namespace Roliha.Problems
{
    internal partial class ProblemA
    {
        static ConsoleReader cr;
        static ConsoleWriter cw;
#if COMPETITIVE
        public ProblemA(string outputFilePath = null, bool solving = true)
#else
        static void Main(string[] args)
#endif
        {
#if COMPETITIVE
            SourceExpander.Expander.Expand(outputFilePath: outputFilePath);
#endif
            if (cw != null)
                cw.Dispose();
            cr = new ConsoleReader();
            cw = new ConsoleWriter();
#if COMPETITIVE
            if (solving)
#endif
                Solve();
            cw.Dispose();
        }
    }
}
#endregion
#region Expanded by https://github.com/kzrnm/SourceExpander
namespace Kzrnm.Competitive.IO{using static Utf8Parser;using _R=ConsoleReader;using MI=System.Runtime.CompilerServices.MethodImplAttribute;public class ConsoleReader{internal const int BufSize=1<<12;internal readonly Stream input;internal readonly Encoding encoding;internal readonly byte[]buf;internal int pos;internal int len;[MI(256)]public ConsoleReader():this(Console.OpenStandardInput(),Console.InputEncoding,BufSize){}[MI(256)]public ConsoleReader(Stream input,Encoding encoding):this(input,encoding,BufSize){}[MI(256)]public ConsoleReader(Stream input,Encoding encoding,int bufferSize){this.input=input;this.encoding=encoding;buf=new byte[bufferSize];}[MI(256)]private void FillEntireNumber(){if((uint)pos>=(uint)buf.Length)FillNextBuffer();while(buf[pos]<=' ')if( ++pos>=len)FillNextBuffer();if(pos+21>=buf.Length&&buf[buf.Length-1]>' ')FillEntireNumberImpl();}private void FillEntireNumberImpl(){buf.AsSpan(pos,len-pos).CopyTo(buf);len-=pos;pos=0;var numberOfBytes=input.Read(buf,len,buf.Length-len);if(numberOfBytes==0)buf[len++]=10;else if(numberOfBytes+len<buf.Length)buf[buf.Length-1]=10;len+=numberOfBytes;}private void FillNextBuffer(){if((len=input.Read(buf,0,buf.Length))==0){buf[0]=10;len=1;}else if(len<buf.Length)buf[buf.Length-1]=10;pos=0;}[MI(256)]internal byte ReadByte(){if(pos>=len)FillNextBuffer();return buf[pos++];}[MI(256)]public T Read<T>(){if(typeof(T)==typeof(int))return(T)(object)Int();if(typeof(T)==typeof(uint))return(T)(object)UInt();if(typeof(T)==typeof(long))return(T)(object)Long();if(typeof(T)==typeof(ulong))return(T)(object)ULong();if(typeof(T)==typeof(double))return(T)(object)Double();if(typeof(T)==typeof(decimal))return(T)(object)Decimal();if(typeof(T)==typeof(char))return(T)(object)Char();if(typeof(T)==typeof(string))return(T)(object)Ascii();return Throw<T>();}static T Throw<T>()=>throw new NotSupportedException(typeof(T).Name);[MI(256)]public int Int(){FillEntireNumber();TryParse(buf.AsSpan(pos),out int v,out int bc);pos+=bc;return v;}[MI(256)]public uint UInt(){FillEntireNumber();TryParse(buf.AsSpan(pos),out uint v,out int bc);pos+=bc;return v;}[MI(256)]public long Long(){FillEntireNumber();TryParse(buf.AsSpan(pos),out long v,out int bc);pos+=bc;return v;}[MI(256)]public ulong ULong(){FillEntireNumber();TryParse(buf.AsSpan(pos),out ulong v,out int bc);pos+=bc;return v;}[MI(256)]public double Double(){FillEntireNumber();TryParse(buf.AsSpan(pos),out double v,out int bc);pos+=bc;return v;}[MI(256)]public decimal Decimal(){FillEntireNumber();TryParse(buf.AsSpan(pos),out decimal v,out int bc);pos+=bc;return v;}[MI(256)]public string String(){var sb=new List<byte>();;byte b;do b=ReadByte();while(b<=' ');do{sb.Add(b);b=ReadByte();}while(' '<b);return encoding.GetString(sb.ToArray());}[MI(256)]public string Ascii(){var sb=new StringBuilder();byte b;do b=ReadByte();while(b<=' ');do{sb.Append((char)b);b=ReadByte();}while(' '<b);return sb.ToString();}[MI(256)]public string Line(){var sb=new List<byte>();byte b;do b=ReadByte();while(b<=' ');do{sb.Add(b);b=ReadByte();}while(b!='\n'&&b!='\r');return encoding.GetString(sb.ToArray());}[MI(256)]public char Char(){byte b;do b=ReadByte();while(b<=' ');return(char)b;}[MI(256)]public int Int0()=>Int()-1;[MI(256)]public uint UInt0()=>UInt()-1;[MI(256)]public long Long0()=>Long()-1;[MI(256)]public ulong ULong0()=>ULong()-1;[MI(256)]public static implicit operator int(_R cr)=>cr.Int();[MI(256)]public static implicit operator uint(_R cr)=>cr.UInt();[MI(256)]public static implicit operator long(_R cr)=>cr.Long();[MI(256)]public static implicit operator ulong(_R cr)=>cr.ULong();[MI(256)]public static implicit operator double(_R cr)=>cr.Double();[MI(256)]public static implicit operator decimal(_R cr)=>cr.Decimal();[MI(256)]public static implicit operator string(_R cr)=>cr.Ascii();public T[]Repeat<T>(int count){var arr=new T[count];for(int i=0;i<arr.Length;i++)arr[i]=Read<T>();return arr;}}}
namespace Kzrnm.Competitive.IO{using _W=ConsoleWriter;using MI=System.Runtime.CompilerServices.MethodImplAttribute;public sealed partial class ConsoleWriter:IDisposable{private const int BufSize=1<<12;private readonly StreamWriter sw;public StreamWriter StreamWriter=>sw;public ConsoleWriter():this(Console.OpenStandardOutput(),Console.OutputEncoding){}public ConsoleWriter(Stream output,Encoding encoding):this(output,encoding,BufSize){}public ConsoleWriter(Stream output,Encoding encoding,int bufferSize){sw=new StreamWriter(output,encoding,bufferSize);}[MI(256)]public void Flush()=>sw.Flush();[MI(256)]public void Dispose()=>Flush();[MI(256)]public _W Write<T>(T v){sw.Write(v.ToString());return this;}[MI(256)]public _W WriteLine(){sw.WriteLine();return this;}[MI(256)]public _W WriteLine<T>(T v){sw.WriteLine(v.ToString());return this;}[MI(256)]public _W WriteLineJoin<T>(IEnumerable<T>col)=>WriteMany(' ',col);[MI(256)]public _W WriteLineJoin<T1,T2>((T1,T2)tuple)=>WriteLineJoin(tuple.Item1,tuple.Item2);[MI(256)]public _W WriteLineJoin<T1,T2,T3>((T1,T2,T3)tuple)=>WriteLineJoin(tuple.Item1,tuple.Item2,tuple.Item3);[MI(256)]public _W WriteLineJoin<T1,T2,T3,T4>((T1,T2,T3,T4)tuple)=>WriteLineJoin(tuple.Item1,tuple.Item2,tuple.Item3,tuple.Item4);[MI(256)]public _W WriteLineJoin<TTuple>(TTuple tuple)where TTuple:System.Runtime.CompilerServices.ITuple{var col=new object[tuple.Length];for(int i=0;i<col.Length;i++)col[i]=tuple[i];return WriteLineJoin(col);}[MI(256)]public _W WriteLineJoin(params object[]col)=>WriteMany(' ',col);[MI(256)]public _W WriteLineJoin<T1,T2>(T1 v1,T2 v2){sw.Write(v1.ToString());sw.Write(' ');sw.WriteLine(v2.ToString());return this;}[MI(256)]public _W WriteLineJoin<T1,T2,T3>(T1 v1,T2 v2,T3 v3){sw.Write(v1.ToString());sw.Write(' ');sw.Write(v2.ToString());sw.Write(' ');sw.WriteLine(v3.ToString());return this;}[MI(256)]public _W WriteLineJoin<T1,T2,T3,T4>(T1 v1,T2 v2,T3 v3,T4 v4){sw.Write(v1.ToString());sw.Write(' ');sw.Write(v2.ToString());sw.Write(' ');sw.Write(v3.ToString());sw.Write(' ');sw.WriteLine(v4.ToString());return this;}[MI(256)]public _W WriteLines<T>(IEnumerable<T>col)=>WriteMany('\n',col);[MI(256)]public _W WriteGrid<T>(IEnumerable<IEnumerable<T>>cols){foreach(var col in cols)WriteLineJoin(col);return this;}[MI(256)]public _W WriteGrid<TTuple>(IEnumerable<TTuple>tuples)where TTuple:System.Runtime.CompilerServices.ITuple{foreach(var tup in tuples)WriteLineJoin(tup);return this;}[MI(256)]private _W WriteMany<T>(char sep,IEnumerable<T>col){var en=col.GetEnumerator();if(!en.MoveNext())goto End;sw.Write(en.Current.ToString());while(en.MoveNext()){sw.Write(sep);sw.Write(en.Current.ToString());}End:sw.WriteLine();return this;}}}
namespace Kzrnm.Competitive.IO{using MI=System.Runtime.CompilerServices.MethodImplAttribute;public partial class ConsoleWriter{[MI(256)]public ConsoleWriter WriteLine(ReadOnlySpan<char>v){sw.WriteLine(v);return this;}[MI(256)]public ConsoleWriter WriteLineJoin<T>(Span<T>col)=>WriteMany(' ',(ReadOnlySpan<T>)col);[MI(256)]public ConsoleWriter WriteLineJoin<T>(ReadOnlySpan<T>col)=>WriteMany(' ',col);[MI(256)]public ConsoleWriter WriteLines<T>(Span<T>col)=>WriteMany('\n',(ReadOnlySpan<T>)col);[MI(256)]public ConsoleWriter WriteLines<T>(ReadOnlySpan<T>col)=>WriteMany('\n',col);[MI(256)]private ConsoleWriter WriteMany<T>(char sep,ReadOnlySpan<T>col){var en=col.GetEnumerator();if(!en.MoveNext())return this;sw.Write(en.Current.ToString());while(en.MoveNext()){sw.Write(sep);sw.Write(en.Current.ToString());}sw.WriteLine();return this;}}}
namespace SourceExpander{public class Expander{[Conditional("EXP")]public static void Expand(string inputFilePath=null,string outputFilePath=null,bool ignoreAnyError=true){}public static string ExpandString(string inputFilePath=null,bool ignoreAnyError=true){return "";}}}
#endregion Expanded by https://github.com/kzrnm/SourceExpander
0