#593760 (C++17) No.1318 ABCD quadruplets

• 問題: No.1318 ABCD quadruplets
• ユーザー: fura
• 提出日時: 2020-12-15 20:01:49
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 618 ms / 2,000 ms
• コード長: 619 bytes
• コンパイル時間: 1,797 ms
• コンパイル使用メモリ: 201,468 KB
• 実行使用メモリ: 43,392 KB
• 最終ジャッジ日時: 2024-09-20 02:05:31
• 合計ジャッジ時間: 8,092 ms
• ジャッジサーバーID （参考情報）: judge4 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	3 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	16 ms; 5,504 KB
testcase_14	AC	254 ms; 15,744 KB
testcase_15	AC	11 ms; 5,376 KB
testcase_16	AC	294 ms; 27,008 KB
testcase_17	AC	56 ms; 5,376 KB
testcase_18	AC	213 ms; 14,848 KB
testcase_19	AC	144 ms; 8,832 KB
testcase_20	AC	15 ms; 5,376 KB
testcase_21	AC	47 ms; 8,064 KB
testcase_22	AC	73 ms; 5,376 KB
testcase_23	AC	590 ms; 42,240 KB
testcase_24	AC	527 ms; 31,360 KB
testcase_25	AC	180 ms; 10,624 KB
testcase_26	AC	135 ms; 7,936 KB
testcase_27	AC	399 ms; 23,168 KB
testcase_28	AC	199 ms; 11,392 KB
testcase_29	AC	241 ms; 13,056 KB
testcase_30	AC	618 ms; 43,392 KB
testcase_31	AC	188 ms; 11,008 KB
testcase_32	AC	597 ms; 43,264 KB

ソースコード

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(n);i++)

using namespace std;

int main(){
	int n,m; scanf("%d%d",&n,&m);

	static int g[490][160001]; // g[i][j] = (a+b+c=i, a^2+ab+ac+b^2+bc+c^2=j となる (a,b,c) の個数)
	rep(a,m+1){
		int x=a*a;
		if(x>n) break;
		rep(b,m+1){
			int y=(a+b)*b;
			if(x+y>n) break;
			rep(c,m+1){
				int z=(a+b+c)*c;
				if(x+y+z>n) break;
				g[a+b+c][x+y+z]++;

			}
		}
	}

	vector<int> ans(n+1);
	rep(i,490) rep(j,n+1) if(g[i][j]>0) {
		rep(d,m+1){
			int w=(i+d)*d;
			if(j+w>n) break;
			ans[j+w]+=g[i][j];
		}
	}
	rep(i,n+1) printf("%d\n",ans[i]);

	return 0;
}