結果

問題 No.2758 RDQ
コンテスト
ユーザー ゆにぽけ
提出日時 2024-05-17 21:33:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 77 ms / 2,000 ms
コード長 2,692 bytes
コンパイル時間 1,390 ms
コンパイル使用メモリ 137,608 KB
最終ジャッジ日時 2025-02-21 14:35:13
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <array>
#include <iterator>
#include <string>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <cassert>
#include <cmath>
#include <ctime>
#include <iomanip>
#include <numeric>
#include <stack>
#include <queue>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <bitset>
#include <random>
#include <utility>
#include <functional>
#include <chrono>
using namespace std;
struct Eratosthenes {
    vector<int> isp,minfactor,myprime,moebius;

    Eratosthenes (int n) : isp(n+1,1),minfactor(n+1,__INT_MAX__),moebius(n+1,1){

        isp[0] = isp[1] = 0;
        minfactor[0] = 0;minfactor[1] = 1;

        for(int p = 2;p <= n;p++){
            if(!isp[p]) continue;
            minfactor[p] = p;
            moebius[p] = -1;
            myprime.push_back(p);

            for(int q = 2*p;q <= n;q += p){
                isp[q] = false;

                if(!((q/p)%p)) moebius[q] = 0;
                else moebius[q] = -moebius[q];


                minfactor[q] = min(minfactor[q],p);
            }
        }
    }

    vector<pair<int,int>> prime_factorize(int n) {
        vector<pair<int,int>> res;
        while(n-1){

            int p = minfactor[n];
            int exp = 0;

            while(minfactor[n] == p){
                n /= p;
                exp ++;
            }

            res.push_back(make_pair(p,exp));
        }

        return res;
    }

    vector<int> divisors(int n){
        vector<int> res;
        res.push_back(1);

        auto pf = prime_factorize(n);

        for(auto [p,e]:pf){
            int s = (int)res.size();

            for(int i = 0;i < s;i++){
                int v = 1;
                for(int j = 0;j < e;j++){
                    v *= p;
                    res.push_back(res[i]*v);
                }
            }
        }

        return res;
    }
};
void Main()
{
	int N,Q;
	cin >> N >> Q;
	vector<int> A(N);
	for(int i = 0;i < N;i++)
	{
		cin >> A[i];
	}
	Eratosthenes E((int)1e5);
	vector<int> L(Q),R(Q),K(Q);
	vector<vector<pair<int,int>>> G(N + 1);
	for(int i = 0;i < Q;i++)
	{
		cin >> L[i] >> R[i] >> K[i];
		L[i]--;
		G[L[i]].push_back(make_pair(i, -1));
		G[R[i]].push_back(make_pair(i, 1));
	}
	vector<int> ans(Q);
	vector<int> cnt((int)1e5 + 1);
	for(int i = 0;i <= N;i++)
	{
		for(const auto &[qi, x] : G[i])
		{
			ans[qi] += x * cnt[K[qi]];
		}
		if(i < N)
		{
			for(int d : E.divisors(A[i]))
			{
				cnt[d]++;
			}
		}
	}
	for(int i = 0;i < Q;i++)
	{
		cout << ans[i] << "\n";
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int tt = 1;
	/* cin >> tt; */
	while(tt--) Main();
}
0