結果

問題 No.854 公平なりんご分配
ユーザー lightning
提出日時 2019-07-26 23:04:41
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,665 bytes
コンパイル時間 2,102 ms
コンパイル使用メモリ 181,516 KB
実行使用メモリ 160,512 KB
最終ジャッジ日時 2024-07-02 09:07:35
合計ジャッジ時間 10,703 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20 WA * 30 TLE * 1 -- * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
#define Rep(i,n) for(int i=0;i<(int)(n);i++)
#define For(i,n1,n2) for(int i=(int)(n1);i<(int)(n2);i++)
#define REP(i,n) for(ll i=0;i<(ll)(n);i++)
#define RREP(i,n) for(ll i=((ll)(n)-1);i>=0;i--)
#define FOR(i,n1,n2) for(ll i=(ll)(n1);i<(ll)(n2);i++)
#define RFOR(i,n1,n2) for(ll i=((ll)(n1)-1);i>=(ll)(n2);i--)
#define put(a) cout<<a<<"\n"
#define all(a)  (a).begin(),(a).end()
#define SORT(a) sort((a).begin(),(a).end())
#define oorret 0
#define oor(x) [&](){try{x;} catch(const out_of_range& oor){return oorret;} return x;}()
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll> P;
template<typename T1, typename T2> inline bool chmin(T1& a, T2 b) { if (a > b) { a = b; return 1; }return 0; }
template<typename T1, typename T2> inline bool chmax(T1& a, T2 b) { if (a < b) { a = b; return 1; }return 0; }

template<typename T>
T upsum(const T& l, const T& r) {//SegmentTreeのupdate関数
	T res;
	res = l + r;
	return res;
}

template<typename T>
class SegmentTree {
public:
	using F = function<T(T&, T&)>;
	vector<T> seg;
	int sz = 1;
	T unit;
	F up;
	//T (*up)(const T& l,const T& r);
	SegmentTree(int n, F up, T unit = 0):up(up),unit(unit) {
		//this->unit = unit;
		//int sz = 1;
		while (sz < n) sz <<= 1;
		seg.resize(sz * 2, unit);
		//this->up = up;
	}
	void set(const int& k, const T& x) {//左からk番目の葉にxを代入する
		seg[sz - 1 + k] = x;
	}
	T get(const int& k) {//左からk番目の葉を得る
		return seg[sz - 1 + k];
	}
	void update(int k, const T& v) {// k 番目の値をvに変更
		k += sz - 1;
		seg[k] = v;
		while (k > 0) {
			k = (k - 1) / 2;
			seg[k] = up(seg[2 * k + 1], seg[2 * k + 2]);
		}
	}
	T query(int a, int b) {//[a,b)のupの合成を求める
		int k = 0, l = 0, r = sz;
		if (r <= a || b <= l) {
			return unit;
		}
		if (a <= l && r <= b) {
			return seg[k];
		}
		else {
			T vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
			T vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
			return up(vl, vr);
		}
	}
	T query(int a, int b, int k, int l, int r) {//[a,b)のupの合成を求める(a,b,0,0,sz)のように使う
		if (r <= a || b <= l) {
			return unit;
		}
		if (a <= l && r <= b) {
			return seg[k];
		}
		else {
			T vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
			T vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
			return up(vl, vr);
		}
	}
};


int main() {
	int n;
	cin >> n;
	vector<int> a(n);
	REP(i, n)cin >> a[i];
	int q;
	cin >> q;
	vector<int> p(q), l(q), r(q);
	REP(i, q)cin >> p[i] >> l[i] >> r[i];
	vector<int> prime;
	FOR(i,2, 2001) {
		bool flag = true;
		for (int j = 2; j * j <= i; ++j) {
			if (i % j == 0) {
				flag = false;
				break;
			}
		}
		if (flag) {
			prime.push_back(i);
		}
	}
	vector<SegmentTree<ll>> sg(prime.size(), SegmentTree<ll>(n, upsum<ll>, 0));
	//SegmentTree<int> sgmax(n, upsum<int>, -1);
	int m = prime.size();
	REP(i, n) {
		int k = a[i];
		for (int j = 0; j < m;) {
			if (k % prime[j] == 0) {
				ll t = sg[j].get(i);
				sg[j].update(i,t + 1);
				k /= prime[j];
			}
			else {
				j++;
			}
			if (k == 1) {
				break;
			}
		}
	}

	REP(i, q) {
		vector<ll> ps(m, 0);
		REP(j, m) {
			ps[j] = sg[j].query(l[i]-1, r[i]);
		}
		vector<ll> pt(m, 0);
		int k = p[i];
		for (int j = 0; j < m;) {
			if (k % prime[j] == 0) {
				pt[j]++;
				k /= prime[j];
			}
			else {
				j++;
			}
			if (k == 1) {
				break;
			}
		}
		if (k != 1) {
			put("NO");
			return 0;
		}
		bool flag = true;
		REP(j, m) {
			if (ps[j] < 0) {
				put(ps.at(10000000000));
			}
			if (ps[j] < pt[j]) {
				flag = false;
				break;
			}
		}
		if (flag) {
			put("Yes");
		}
		else {
			put("NO");
		}
	}
 	return 0;
}
0