結果

問題 No.3107 45^2
ユーザー mutv625
提出日時 2025-04-18 20:03:48
言語 C#
(.NET 8.0.404)
結果
AC  
実行時間 54 ms / 2,000 ms
コード長 14,597 bytes
コンパイル時間 8,909 ms
コンパイル使用メモリ 169,512 KB
実行使用メモリ 189,240 KB
最終ジャッジ日時 2025-04-18 20:03:58
合計ジャッジ時間 9,068 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
  復元対象のプロジェクトを決定しています...
  /home/judge/data/code/main.csproj を復元しました (122 ミリ秒)。
/home/judge/data/code/Main.cs(317,108): warning CS8632: '#nullable' 注釈コンテキスト内のコードでのみ、Null 許容参照型の注釈を使用する必要があります。 [/home/judge/data/code/main.csproj]
/home/judge/data/code/Main.cs(272,109): warning CS8632: '#nullable' 注釈コンテキスト内のコードでのみ、Null 許容参照型の注釈を使用する必要があります。 [/home/judge/data/code/main.csproj]
  main -> /home/judge/data/code/bin/Release/net8.0/main.dll
  main -> /home/judge/data/code/bin/Release/net8.0/publish/

ソースコード

diff #

using System.Collections;
using System.Numerics;
using System.Runtime.CompilerServices;
using System.Runtime.InteropServices;
using System.Runtime.Intrinsics;

using static InputUtil;

// VVVVVVVVVVV ここからコード VVVVVVVVVVV

var n = ReadLine<int>();
Console.WriteLine(n*n);

// XXXXXXXXXXX ここまでコード XXXXXXXXXXXX

#region library
#pragma warning disable CA1050

/// <summary>
/// 標準入力の読み取りを簡素化する機能を提供します。
/// </summary>
public static class InputUtil
{
    /// <summary>
    /// 標準入力からスペースで区切られた値を指定の型に変換して読み込みます。
    /// </summary>
    /// <remarks>
    /// 1つの値を読み込むには型を明示し <c>int a = ReadLine&lt;int&gt;()</c> のように書きます。
    /// 定数個の値を読み取るには <c>var (a, b, c) = ReadLine&lt;int&gt;();</c> のように書きます。
    /// 可変個の値を読み取るには <c>foreach (var item in ReadLine&lt;int&gt;())</c> のように書きます。
    /// <c>ToArray()</c> メソッドや Linq も使えます。
    /// </remarks>
    /// <typeparam name="T">変換先の型</typeparam>
    /// <returns>標準入力の読み取り結果。</returns>
    /// <exception cref="FormatException"></exception>
    /// <exception cref="InvalidOperationException"></exception>
    public static InputToken<T, SpanParsableImpl<T>> ReadLine<T>() where T : ISpanParsable<T>
    {
        return new InputToken<T, SpanParsableImpl<T>>(Console.ReadLine() ?? throw new InvalidOperationException());
    }

    /// <summary>
    /// 標準入力から指定の文字で区切られた値を指定の型に変換して読み込みます。
    /// 使用方法は <see cref="ReadLine{T}"/> を参照してください。
    /// </summary>
    /// <typeparam name="T"></typeparam>
    /// <param name="formatProvider"></param>
    /// <param name="separator"></param>
    /// <returns></returns>
    /// <exception cref="InvalidOperationException"></exception>
    public static InputToken<T, SpanParsableImpl<T>> ReadLine<T>(IFormatProvider formatProvider, char separator) where T : ISpanParsable<T>
    {
        return new InputToken<T, SpanParsableImpl<T>>(Console.ReadLine() ?? throw new InvalidOperationException(), formatProvider, separator);
    }

    /// <summary>
    /// 標準入力からスペースで区切られた文字列を読み込みます。
    /// 使用方法は <see cref="ReadLine{T}"/> を参照してください。
    /// </summary>
    /// <returns></returns>
    /// <exception cref="InvalidOperationException"></exception>
    public static InputToken<string, StringImpl> ReadLine()
    {
        return new InputToken<string, StringImpl>(Console.ReadLine() ?? throw new InvalidOperationException());
    }

    /// <summary>
    /// 標準入力から指定の文字で区切られた文字列を読み込みます。
    /// 使用方法は <see cref="ReadLine{T}"/> を参照してください。
    /// </summary>
    /// <param name="formatProvider"></param>
    /// <param name="separator"></param>
    /// <returns></returns>
    /// <exception cref="InvalidOperationException"></exception>
    public static InputToken<string, StringImpl> ReadLine(IFormatProvider formatProvider, char separator)
    {
        return new InputToken<string, StringImpl>(Console.ReadLine() ?? throw new InvalidOperationException(), formatProvider, separator);
    }
}

public interface IReadNextValueImpl<TResult>
{
    public static abstract TResult ReadNextValue(string input, ref int currentIndex, char separator, IFormatProvider formatProvider);

    public static abstract bool TryReadNextValue(string input, ref int currentIndex, char separator, IFormatProvider formatProvider, out TResult value);
}

public readonly struct SpanParsableImpl<T> : IReadNextValueImpl<T> where T : ISpanParsable<T>
{
    public static T ReadNextValue(string input, ref int currentIndex, char separator, IFormatProvider formatProvider)
    {
        ReadOnlySpan<char> span = input.AsSpan()[currentIndex..];
        var index = span.IndexOf(separator);
        ReadOnlySpan<char> target = index == -1 ? span : span[..index];
        if (target.Length == 0)
        {
            ThrowSeparatorNotFound();
        }
        var value = T.Parse(target, formatProvider);
        currentIndex = index == -1 ? input.Length : currentIndex + index + 1;
        return value;
    }

    private static void ThrowSeparatorNotFound()
    {
        throw new FormatException($"Separator not found");
    }

    public static bool TryReadNextValue(string input, ref int currentIndex, char separator, IFormatProvider formatProvider, out T value)
    {
        ReadOnlySpan<char> span = input.AsSpan()[currentIndex..];
        var index = span.IndexOf(separator);
        ReadOnlySpan<char> target = index == -1 ? span : span[..index];
        if (target.Length == 0)
        {
            value = default!;
            return false;
        }
        value = T.Parse(target, formatProvider);
        currentIndex = index == -1 ? input.Length : currentIndex + index + 1;
        return true;
    }
}

public readonly struct StringImpl : IReadNextValueImpl<string>
{
    public static string ReadNextValue(string input, ref int currentIndex, char separator, IFormatProvider formatProvider)
    {
        ReadOnlySpan<char> span = input.AsSpan()[currentIndex..];
        var index = span.IndexOf(separator);
        ReadOnlySpan<char> target = index == -1 ? span : span[..index];
        if (target.Length == 0)
        {
            ThrowSeparatorNotFound();
        }
        var value = target.ToString();
        currentIndex = index == -1 ? input.Length : currentIndex + index + 1;
        return value;
    }

    private static void ThrowSeparatorNotFound()
    {
        throw new FormatException($"Separator not found");
    }

    public static bool TryReadNextValue(string input, ref int currentIndex, char separator, IFormatProvider formatProvider, out string value)
    {
        ReadOnlySpan<char> span = input.AsSpan()[currentIndex..];
        var index = span.IndexOf(separator);
        ReadOnlySpan<char> target = index == -1 ? span : span[..index];
        if (target.Length == 0)
        {
            value = default!;
            return false;
        }
        value = target.ToString();
        currentIndex = index == -1 ? input.Length : currentIndex + index + 1;
        return true;
    }
}

public struct InputToken<TResult, TImpl> : IEnumerable<TResult>, IEnumerator<TResult> where TImpl : IReadNextValueImpl<TResult>
{
    private readonly string _input;

    private int _currentIndex;

    private readonly IFormatProvider _formatProvider;

    private readonly char _separator;

#pragma warning disable CS8618 // Property Current is initialized in MoveNext
    public InputToken(string input)
    {
        _input = input;
        _formatProvider = System.Globalization.CultureInfo.InvariantCulture;
        _separator = ' ';
    }

    public InputToken(string input, IFormatProvider formatProvider, char separator)
    {
        _input = input;
        _formatProvider = formatProvider;
        _separator = separator;
    }
#pragma warning restore CS8618

    public static implicit operator TResult(InputToken<TResult, TImpl> token)
    {
        return token.ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2, out TResult val3)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
        val3 = ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2, out TResult val3, out TResult val4)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
        val3 = ReadNextValue();
        val4 = ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2, out TResult val3, out TResult val4, out TResult val5)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
        val3 = ReadNextValue();
        val4 = ReadNextValue();
        val5 = ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2, out TResult val3, out TResult val4, out TResult val5, out TResult val6)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
        val3 = ReadNextValue();
        val4 = ReadNextValue();
        val5 = ReadNextValue();
        val6 = ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2, out TResult val3, out TResult val4, out TResult val5, out TResult val6, out TResult val7)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
        val3 = ReadNextValue();
        val4 = ReadNextValue();
        val5 = ReadNextValue();
        val6 = ReadNextValue();
        val7 = ReadNextValue();
    }

    public void Deconstruct(out TResult val1, out TResult val2, out TResult val3, out TResult val4, out TResult val5, out TResult val6, out TResult val7, out TResult val8)
    {
        val1 = ReadNextValue();
        val2 = ReadNextValue();
        val3 = ReadNextValue();
        val4 = ReadNextValue();
        val5 = ReadNextValue();
        val6 = ReadNextValue();
        val7 = ReadNextValue();
        val8 = ReadNextValue();
    }

    private TResult ReadNextValue()
    {
        return TImpl.ReadNextValue(_input, ref _currentIndex, _separator, _formatProvider);
    }

    public readonly InputToken<TResult, TImpl> GetEnumerator()
    {
        return this;
    }

    public TResult Current { get; private set; }

    readonly object IEnumerator.Current => Current!;

    public bool MoveNext()
    {
        var val = TImpl.TryReadNextValue(_input, ref _currentIndex, _separator, _formatProvider, out TResult? current);
        Current = current;
        return val;
    }

    public TResult[] ToArray()
    {
        if (_input.Length == 0)
        {
            return Array.Empty<TResult>();
        }
        ref var begin = ref Unsafe.As<char, ushort>(ref MemoryMarshal.GetReference(_input.AsSpan()));
        var count = SpanHelpers.CountValueType(ref begin, _separator, _input.Length) + (_input[^1] == _separator ? 0 : 1);
        var array = new TResult[count];
        for (var i = 0; i < array.Length; i++)
        {
            array[i] = ReadNextValue();
        }
        return array;
    }

    readonly IEnumerator<TResult> IEnumerable<TResult>.GetEnumerator()
    {
        return this;
    }

    readonly IEnumerator IEnumerable.GetEnumerator()
    {
        return this;
    }

    void IEnumerator.Reset()
    {
        _currentIndex = 0;
    }

    readonly void IDisposable.Dispose()
    {

    }
}

// Source: [System.Private.CoreLib]System.SpanHelpers
internal static class SpanHelpers
{
    public static int CountValueType<T>(ref T current, T value, int length) where T : struct, IEquatable<T>?
    {
        var count = 0;
        ref T end = ref Unsafe.Add(ref current, length);

        if (Vector128.IsHardwareAccelerated && length >= Vector128<T>.Count)
        {
            // Vector512 is not supported on .NET 7.0 so Vector512 code is not included.
            if (Vector256.IsHardwareAccelerated && length >= Vector256<T>.Count)
            {
                var targetVector = Vector256.Create(value);
                ref T oneVectorAwayFromEnd = ref Unsafe.Subtract(ref end, Vector256<T>.Count);
                do
                {
                    count += BitOperations.PopCount(Vector256.Equals(Vector256.LoadUnsafe(ref current), targetVector).ExtractMostSignificantBits());
                    current = ref Unsafe.Add(ref current, Vector256<T>.Count);
                }
                while (!Unsafe.IsAddressGreaterThan(ref current, ref oneVectorAwayFromEnd));

                // If there are just a few elements remaining, then processing these elements by the scalar loop
                // is cheaper than doing bitmask + popcount on the full last vector. To avoid complicated type
                // based checks, other remainder-count based logic to determine the correct cut-off, for simplicity
                // a half-vector size is chosen (based on benchmarks).
                var remaining = (uint)Unsafe.ByteOffset(ref current, ref end) / (uint)Unsafe.SizeOf<T>();
                if (remaining > Vector256<T>.Count / 2)
                {
                    var mask = Vector256.Equals(Vector256.LoadUnsafe(ref oneVectorAwayFromEnd), targetVector).ExtractMostSignificantBits();

                    // The mask contains some elements that may be double-checked, so shift them away in order to get the correct pop-count.
                    var overlaps = (uint)Vector256<T>.Count - remaining;
                    mask >>= (int)overlaps;
                    count += BitOperations.PopCount(mask);

                    return count;
                }
            }
            else
            {
                var targetVector = Vector128.Create(value);
                ref T oneVectorAwayFromEnd = ref Unsafe.Subtract(ref end, Vector128<T>.Count);
                do
                {
                    count += BitOperations.PopCount(Vector128.Equals(Vector128.LoadUnsafe(ref current), targetVector).ExtractMostSignificantBits());
                    current = ref Unsafe.Add(ref current, Vector128<T>.Count);
                }
                while (!Unsafe.IsAddressGreaterThan(ref current, ref oneVectorAwayFromEnd));

                var remaining = (uint)Unsafe.ByteOffset(ref current, ref end) / (uint)Unsafe.SizeOf<T>();
                if (remaining > Vector128<T>.Count / 2)
                {
                    var mask = Vector128.Equals(Vector128.LoadUnsafe(ref oneVectorAwayFromEnd), targetVector).ExtractMostSignificantBits();

                    // The mask contains some elements that may be double-checked, so shift them away in order to get the correct pop-count.
                    var overlaps = (uint)Vector128<T>.Count - remaining;
                    mask >>= (int)overlaps;
                    count += BitOperations.PopCount(mask);

                    return count;
                }
            }
        }

        while (Unsafe.IsAddressLessThan(ref current, ref end))
        {
            if (current.Equals(value))
            {
                count++;
            }

            current = ref Unsafe.Add(ref current, 1);
        }

        return count;
    }
}

#endregion
0