結果

問題 No.2189 六平方和
ユーザー SSRS
提出日時 2023-01-13 22:01:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 999 bytes
コンパイル時間 1,965 ms
コンパイル使用メモリ 167,936 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-24 17:30:11
合計ジャッジ時間 4,229 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 16 WA * 12
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
long long modpow(long long a, long long b, long long m){
  long long ans = 1;
  while (b > 0){
    if (b % 2 == 1){
      ans *= a;
      ans %= m;
    }
    a *= a;
    a %= m;
    b /= 2;
  }
  return ans;
}
int main(){
  long long N;
  int M, B;
  cin >> N >> M >> B;
  long long x = modpow(M, N / 2, B);
  if (x == 0){
    x = B;
  }
  if (N % 2 == 0){
    cout << "YES" << endl;
    cout << x << ' ' << B << ' ' << B << ' ' << B << ' ' << B << ' ' << B << endl;
  } else {
    vector<int> C(6, 0);
    for (int i = 0; i < 6; i++){
      for (int j = M; j >= 0; j--){
        if (j * j <= M){
          C[i] = j * j;
          M -= j * j;
          break;
        }
      }
    }
    assert(M == 0);
    cout << "YES" << endl;
    for (int i = 0; i < 6; i++){
      long long ans = (long long) C[i] * x % B;
      if (ans == 0){
        ans = B;
      }
      cout << ans;
      if (i < 5){
        cout << ' ';
      }
    }
    cout << endl;
  }
}
0