結果
問題 |
No.991 N×Mマス計算(構築)
|
ユーザー |
![]() |
提出日時 | 2020-03-14 02:41:31 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,925 bytes |
コンパイル時間 | 1,850 ms |
コンパイル使用メモリ | 176,644 KB |
実行使用メモリ | 814,768 KB |
最終ジャッジ日時 | 2024-11-23 18:09:51 |
合計ジャッジ時間 | 15,059 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | WA * 2 |
other | WA * 3 RE * 23 TLE * 1 MLE * 1 |
ソースコード
#include<bits/stdc++.h> using namespace std; #define fs first #define sc second #define pb push_back #define mp make_pair #define eb emplace_back #define ALL(A) A.begin(),A.end() #define RALL(A) A.rbegin(),A.rend() typedef long long LL; typedef pair<int,int> P; template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; } template<typename T> T gcd(T a,T b){return b?gcd(b,a%b):a;} const LL mod=1000000007; const LL LINF=1LL<<62; const int INF=1<<30; int dx[]={1,0,-1,0,1,-1,1,-1}; int dy[]={0,1,0,-1,1,-1,-1,1}; int main(){ LL n,m,k;cin >> n >> m >> k; char op;cin >> op; vector<LL> b(m),a(n); for (int i = 0; i < m; i++) { cin >> b[i]; b[i] %= k; } for (int i = 0; i < n; i++) { cin >> a[i]; a[i] %= k; } if(op == '+'){ map<int,LL> B; for (int i = 0; i < m; i++) { B[b[i]]++; } LL ans = 0; for (int i = 0; i < n; i++) { ans += B[(k - a[i])%k]; } cout << ans << endl; } else{ for (int i = 0; i < m; i++) { b[i] = gcd(k, b[i]); if(b[i] == k) b[i] = 0; } for (int i = 0; i < n; i++) { a[i] = gcd(k, a[i]); if(a[i] == k) a[i] = 0; } LL cntb = 0, cnta = 0; map<int,LL> B; LL ans = 0; for (int i = 0; i < m; i++) { if(b[i] == 0) cntb++; else B[k / gcd(k,b[i])]++; } for (int i = 0; i < n; i++) { if(a[i] == 0) cnta++; else{ for(auto m : B){ if(a[i] % m.fs == 0) ans += m.sc; } } } ans += cnta * (m - cntb) + cntb * (n - cnta) - cnta * cntb; cout << ans << endl; } return 0; }