結果

問題 No.800 四平方定理
ユーザー EmKjp
提出日時 2019-05-01 14:14:32
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 221 ms / 2,000 ms
コード長 3,784 bytes
コンパイル時間 1,586 ms
コンパイル使用メモリ 112,380 KB
実行使用メモリ 77,740 KB
最終ジャッジ日時 2024-12-31 12:24:39
合計ジャッジ時間 6,006 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff # 

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


partial class Solver
{
    public void Run()
    {
        var N = ni();
        var D = ni();
        // x^2 + y^2 + z^2 = W^2 + D
        // x^2 + y^2 = W^2 - z^2 + D
        var hist1 = new int[2 * (N + 1) * (N + 1) + 1];
        var hist2 = new int[2 * (N + 1) * (N + 1) + 1];
        for (int i = 1; i <= N && i * i <= hist1.Length; i++)
        {
            for (int j = 1; j <= N; j++)
            {
                if (i * i + j * j <= hist1.Length)
                {
                    hist1[i * i + j * j]++;
                }
                else
                {
                    break;
                }
            }
            for (int j = 1; j <= N; j++)
            {
                var v = i * i - j * j + D;
                if (0 <= v && v < hist2.Length)
                {
                    hist2[v]++;
                }
            }
        }
        long ans = 0;
        for (int i = 0; i < hist2.Length; i++)
        {
            ans += 1L * hist1[i] * hist2[i];
        }
        cout.WriteLine(ans);
    }
}

// PREWRITEN CODE BEGINS FROM HERE
partial class Solver : Scanner
{
    public static void Main(string[] args)
    {
        new Solver(Console.In, Console.Out).Run();
    }

    TextReader cin;
    TextWriter cout;

    public Solver(TextReader reader, TextWriter writer)
        : base(reader)
    {
        this.cin = reader;
        this.cout = writer;
    }
    public Solver(string input, TextWriter writer)
        : this(new StringReader(input), writer)
    {
    }

    public int ni() { return NextInt(); }
    public int[] ni(int n) { return NextIntArray(n); }
    public long nl() { return NextLong(); }
    public long[] nl(int n) { return NextLongArray(n); }
    public double nd() { return NextDouble(); }
    public string ns() { return Next(); }
    public string[] ns(int n) { return NextArray(n); }
}

public class Scanner
{
    private TextReader Reader;
    private Queue<String> TokenQueue = new Queue<string>();
    private CultureInfo ci = CultureInfo.InvariantCulture;

    public Scanner()
        : this(Console.In)
    {
    }

    public Scanner(TextReader reader)
    {
        this.Reader = reader;
    }

    public int NextInt() { return Int32.Parse(Next(), ci); }
    public long NextLong() { return Int64.Parse(Next(), ci); }
    public double NextDouble() { return double.Parse(Next(), ci); }
    public string[] NextArray(int size)
    {
        var array = new string[size];
        for (int i = 0; i < size; i++) array[i] = Next();
        return array;
    }
    public int[] NextIntArray(int size)
    {
        var array = new int[size];
        for (int i = 0; i < size; i++) array[i] = NextInt();
        return array;
    }

    public long[] NextLongArray(int size)
    {
        var array = new long[size];
        for (int i = 0; i < size; i++) array[i] = NextLong();
        return array;
    }

    public String Next()
    {
        if (TokenQueue.Count == 0)
        {
            if (!StockTokens()) throw new InvalidOperationException();
        }
        return TokenQueue.Dequeue();
    }

    public bool HasNext()
    {
        if (TokenQueue.Count > 0)
            return true;
        return StockTokens();
    }

    private bool StockTokens()
    {
        while (true)
        {
            var line = Reader.ReadLine();
            if (line == null) return false;
            var tokens = line.Trim().Split(" ".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
            if (tokens.Length == 0) continue;
            foreach (var token in tokens)
                TokenQueue.Enqueue(token);
            return true;
        }
    }
}
0