結果

問題 No.800 四平方定理
ユーザー kitakitalilykitakitalily
提出日時 2019-04-02 23:00:13
言語 Java21
(openjdk 21)
結果
RE  
実行時間 -
コード長 4,609 bytes
コンパイル時間 2,577 ms
コンパイル使用メモリ 91,396 KB
実行使用メモリ 101,184 KB
最終ジャッジ日時 2024-05-08 09:30:57
合計ジャッジ時間 10,997 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 193 ms
100,292 KB
testcase_01 AC 188 ms
99,848 KB
testcase_02 AC 188 ms
100,020 KB
testcase_03 AC 184 ms
100,064 KB
testcase_04 AC 187 ms
99,764 KB
testcase_05 AC 180 ms
99,552 KB
testcase_06 AC 184 ms
100,136 KB
testcase_07 AC 189 ms
100,716 KB
testcase_08 AC 179 ms
100,076 KB
testcase_09 AC 186 ms
99,956 KB
testcase_10 AC 222 ms
100,356 KB
testcase_11 AC 219 ms
99,968 KB
testcase_12 AC 230 ms
100,336 KB
testcase_13 AC 227 ms
100,228 KB
testcase_14 AC 233 ms
100,520 KB
testcase_15 AC 215 ms
99,924 KB
testcase_16 AC 211 ms
99,996 KB
testcase_17 AC 219 ms
100,284 KB
testcase_18 AC 215 ms
100,360 KB
testcase_19 AC 224 ms
100,184 KB
testcase_20 AC 180 ms
100,180 KB
testcase_21 AC 193 ms
99,800 KB
testcase_22 AC 220 ms
99,936 KB
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 AC 180 ms
100,232 KB
testcase_27 AC 184 ms
100,180 KB
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
権限があれば一括ダウンロードができます

ソースコード

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++){
        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