結果

問題 No.1349 Subset Product Queries
ユーザー ninhtay
提出日時 2025-06-22 04:02:33
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,752 bytes
コンパイル時間 2,252 ms
コンパイル使用メモリ 204,480 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-06-22 04:02:41
合計ジャッジ時間 8,280 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 5 WA * 5 RE * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

const int MAX_N = 1005;
const int MAX_P = 105;
const int MAX_Q = 100005;
using ll = long long;

int N, Q, P;
vector<ll> A(MAX_N);
vector<int> L(MAX_Q), R(MAX_Q), K(MAX_Q);
vector<vector<bool>> dp(MAX_N, vector<bool>(MAX_P, false));

void preprocess() {
    dp[0][1] = true; 
    for (int i = 1; i <= N; i++) {
        
        for (int k = 0; k < P; k++) {
            dp[i][k] = dp[i][k] |dp[i-1][k];
        }
        
        for (int k = 0; k < P; k++) {
            if (dp[i-1][k]) {
                dp[i][(k * A[i]) % P] = true;
            }
        }
    }
}


bool check_segment(int left, int right, int K) {
    
    if (!dp[right][K]) return false;
    
    if (left == 1) return true;
    
    vector<vector<bool>> dp2(right - left + 2, vector<bool>(P, false));
    dp2[0][1] = true; 
    for (int i = left; i <= right; i++) {
        int idx = i - left + 1;
        
        for (int k = 0; k < P; k++) {
            dp2[idx][k] = dp2[idx][k]|dp2[idx-1][k];
        }
        
        for (int k = 0; k < P; k++) {
            if (dp2[idx-1][k]) {
                dp2[idx][(k * A[i]) % P] = true;
            }
        }
    }
    return dp2[right - left + 1][K];
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    
    cin >> N >> Q >> P;
    for (int i = 1; i <= N; i++) {
        cin >> A[i];
    }
    for (int i = 1; i <= Q; i++) {
        cin >> L[i] >> R[i] >> K[i];
    }

    
    preprocess();

    
    int valid_queries = 0;
    for (int i = 1; i <= Q; i++) {
        if (check_segment(L[i], R[i], K[i])) {
            cout<<"Yes"<<endl;
        }
        else
            cout<<"No"<<endl;
    }

    
   // cout << valid_queries << endl;

    return 0;
}
0