結果

問題 No.800 四平方定理
ユーザー kitakitalilykitakitalily
提出日時 2019-04-02 23:01:34
言語 Java21
(openjdk 21)
結果
AC  
実行時間 245 ms / 2,000 ms
コード長 4,655 bytes
コンパイル時間 4,663 ms
コンパイル使用メモリ 76,332 KB
実行使用メモリ 102,944 KB
最終ジャッジ日時 2023-08-21 04:02:31
合計ジャッジ時間 11,290 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 165 ms
101,944 KB
testcase_01 AC 159 ms
102,192 KB
testcase_02 AC 159 ms
101,972 KB
testcase_03 AC 161 ms
101,728 KB
testcase_04 AC 159 ms
102,056 KB
testcase_05 AC 162 ms
102,116 KB
testcase_06 AC 161 ms
101,724 KB
testcase_07 AC 161 ms
102,340 KB
testcase_08 AC 164 ms
102,204 KB
testcase_09 AC 161 ms
101,888 KB
testcase_10 AC 200 ms
102,800 KB
testcase_11 AC 189 ms
102,944 KB
testcase_12 AC 189 ms
102,804 KB
testcase_13 AC 193 ms
102,788 KB
testcase_14 AC 188 ms
102,400 KB
testcase_15 AC 191 ms
102,564 KB
testcase_16 AC 192 ms
102,808 KB
testcase_17 AC 192 ms
102,684 KB
testcase_18 AC 206 ms
102,528 KB
testcase_19 AC 199 ms
102,672 KB
testcase_20 AC 158 ms
102,392 KB
testcase_21 AC 156 ms
102,088 KB
testcase_22 AC 193 ms
102,540 KB
testcase_23 AC 242 ms
102,644 KB
testcase_24 AC 237 ms
102,452 KB
testcase_25 AC 245 ms
102,708 KB
testcase_26 AC 158 ms
102,108 KB
testcase_27 AC 164 ms
101,888 KB
testcase_28 AC 238 ms
102,356 KB
testcase_29 AC 241 ms
102,596 KB
testcase_30 AC 232 ms
102,652 KB
testcase_31 AC 227 ms
102,348 KB
testcase_32 AC 232 ms
102,224 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