結果

問題 No.800 四平方定理
ユーザー kitakitalilykitakitalily
提出日時 2019-04-02 23:01:34
言語 Java21
(openjdk 21)
結果
AC  
実行時間 331 ms / 2,000 ms
コード長 4,655 bytes
コンパイル時間 2,800 ms
コンパイル使用メモリ 80,068 KB
実行使用メモリ 90,240 KB
最終ジャッジ日時 2024-05-08 09:33:11
合計ジャッジ時間 12,489 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 205 ms
89,704 KB
testcase_01 AC 205 ms
89,552 KB
testcase_02 AC 205 ms
89,504 KB
testcase_03 AC 208 ms
89,900 KB
testcase_04 AC 201 ms
89,532 KB
testcase_05 AC 208 ms
89,672 KB
testcase_06 AC 209 ms
89,804 KB
testcase_07 AC 207 ms
89,632 KB
testcase_08 AC 211 ms
89,544 KB
testcase_09 AC 205 ms
89,728 KB
testcase_10 AC 270 ms
90,012 KB
testcase_11 AC 256 ms
89,856 KB
testcase_12 AC 262 ms
89,852 KB
testcase_13 AC 255 ms
89,800 KB
testcase_14 AC 249 ms
89,680 KB
testcase_15 AC 243 ms
89,692 KB
testcase_16 AC 247 ms
89,800 KB
testcase_17 AC 249 ms
89,760 KB
testcase_18 AC 246 ms
89,492 KB
testcase_19 AC 250 ms
89,640 KB
testcase_20 AC 202 ms
90,068 KB
testcase_21 AC 203 ms
89,860 KB
testcase_22 AC 249 ms
89,720 KB
testcase_23 AC 331 ms
89,744 KB
testcase_24 AC 312 ms
89,804 KB
testcase_25 AC 323 ms
90,240 KB
testcase_26 AC 203 ms
89,712 KB
testcase_27 AC 204 ms
89,628 KB
testcase_28 AC 316 ms
90,240 KB
testcase_29 AC 314 ms
89,736 KB
testcase_30 AC 305 ms
89,932 KB
testcase_31 AC 304 ms
90,188 KB
testcase_32 AC 310 ms
90,136 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.Scanner;
import java.util.Collections;
import java.util.List;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Queue;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.PriorityQueue;
import java.util.Set;
import java.util.Map;
import java.util.HashMap;
import java.util.TreeSet;
import java.util.Comparator;

public class Main {
	static int mod = 1000000007;
  static int size = 200010;
	static long[] fac = new long[size];
	static long[] finv = new long[size];
	static long[] inv = new long[size];
  static int INF = Integer.MAX_VALUE;

 	public static void main(String[] args){
		Scanner scanner = new Scanner(System.in);
    int n = scanner.nextInt();
    int d = scanner.nextInt();
    int[] cnt1 = new int[5000001];
    int[] cnt2 = new int[5000001];
    for(int i = 1; i <= n; i++){
      for(int j = 1; j <= n; j++){
        if(i*i+j*j <= 5000000){
          cnt1[i*i+j*j]++;
        }  
      }
    }
    for(int i = 1; i <= n; i++){
      for(int j = 1; j <= n; j++){
        if(i*i-j*j+d>=0){
          cnt2[i*i-j*j+d]++;
        }
      }
    }
    int ans = 0;
    for(int i = 1; i <= 5000000; i++){
      ans += cnt1[i] * cnt2[i];
    }
    System.out.println(ans);
  }
  static class Pair implements Comparable<Pair>{
    int x, y;
    Pair(int a, int b){
        x = a;
        y = b;
    }
    @Override
    public boolean equals(Object o){
        if (this == o) return true;
        if (!(o instanceof Pair)) return false;
        Pair p = (Pair) o;
        return x == p.x && y == p.y;
    }
    @Override
    public int compareTo(Pair p){
        return x == p.x ? y - p.y : x - p.x; //xで昇順にソート
        //return (x == p.x ? y - p.y : x - p.x) * -1; //xで降順にソート
        //return y == p.y ? x - p.x : y - p.y;//yで昇順にソート
        //return (y == p.y ? x - p.x : y - p.y)*-1;//yで降順にソート
    }
  }
  static class Edge implements Comparable<Edge>{
    int from;
    int to;
    int cost;

    Edge(int from, int to,int cost){
      this.from = from;
      this.to = to;
      this.cost = cost;
    }

    public int compareTo(Edge e) {
      return this.cost - e.cost;
    }

  }
  //繰り返し二乗法
  public static long pow(long x, long n){
    long ans = 1;
    while(n > 0){
      if((n & 1) == 1){
        ans = ans * x;
        ans %= mod;
      }
      x = x * x % mod;
      n >>= 1;
    }
    return ans;
  }

  //fac, inv, finvテーブルの初期化、これ使う場合はinitComb()で初期化必要
	public static  void initComb(){
		fac[0] = finv[0] = inv[0] = fac[1] = finv[1] = inv[1] = 1;
		for (int i = 2; i < size; ++i) {
			fac[i] = fac[i - 1] * i % mod;
			inv[i] = mod - (mod / i) * inv[(int) (mod % i)] % mod;
			finv[i] = finv[i - 1] * inv[i] % mod;
		}
	}

	//nCk % mod
	public static  long comb(int n, int k){
		return fac[n] * finv[k] % mod * finv[n - k] % mod;
	}

	//n! % mod
	public static  long fact(int n){
		return fac[n];
	}

	//(n!)^-1 with % mod
	public static long finv(int n){
		return finv[n];
	}

  static class UnionFind {
    int[] parent;
    public UnionFind(int size) {
      parent = new int[size];
      Arrays.fill(parent, -1);
    }
    public boolean unite(int x, int y) {
      x = root(x);
      y = root(y);
      if (x != y) {
        if (parent[y] < parent[x]) {
          int tmp = y;
          y = x;
          x = tmp;
        }
        parent[x] += parent[y];
        parent[y] = x;
        return true;
      }
      return false;
    }
    public boolean same(int x, int y) {
      return root(x) == root(y);
    }
    public int root(int x) {
      return parent[x] < 0 ? x : (parent[x] = root(parent[x]));
    }
    public int size(int x) {
      return -parent[root(x)];
    }
  }
  public static int upperBound(long[] a,long val){
    return upperBound(a,0,a.length,val);
  }
  public static int upperBound(long[] a,int l,int r,long val){
    if(r-l==1){
      if(a[l]>val) return l;
      return r;
    }
    int mid=(l+r)/2;
    if(a[mid]>val){
      return upperBound(a,l,mid,val);
    }else{
      return upperBound(a,mid,r,val);
    }
  }
  public static int lowerBound(long[] a,long val){
     return lowerBound(a,0,a.length,val);
 }
  public static int lowerBound(long[] a,int l,int r,long val){
    if(r-l==1){
      if(a[l]<val) return r;
      return l;
    }
    int mid=(l+r)/2;
    if(a[mid]<val){
      return lowerBound(a,mid,r,val);
    }else{
      return lowerBound(a,l,mid,val);
    }
  }
  //n,mの最大公約数
  public static int gcd(int n, int m){
    if(m > n) return gcd(m,n);
    if(m == 0) return n;
    return gcd(m, n%m);
  }
}
0