#189036 (C++14) No.546 オンリー・ワン

• 問題: No.546 オンリー・ワン
• ユーザー: gzlcp
• 提出日時: 2017-07-14 23:38:06
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 5 ms / 2,000 ms
• コード長: 1,102 bytes
• コンパイル時間: 837 ms
• コンパイル使用メモリ: 87,548 KB
• 実行使用メモリ: 6,820 KB
• 最終ジャッジ日時: 2024-10-07 23:44:06
• 合計ジャッジ時間: 1,497 ms
• ジャッジサーバーID （参考情報）: judge5 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 6,820 KB
testcase_01	AC	2 ms; 6,816 KB
testcase_02	AC	2 ms; 6,816 KB
testcase_03	AC	2 ms; 6,816 KB
testcase_04	AC	2 ms; 6,820 KB
testcase_05	AC	2 ms; 6,820 KB
testcase_06	AC	2 ms; 6,820 KB
testcase_07	AC	2 ms; 6,816 KB
testcase_08	AC	3 ms; 6,820 KB
testcase_09	AC	5 ms; 6,820 KB
testcase_10	AC	2 ms; 6,816 KB

ソースコード

#include<iostream>
#include<math.h>
#include<vector>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#define REP(i, n) for(long long i = 0; i < n; ++i)
#define ALL(a) a.begin(), a.end()
#define INF (long long)1000000000
using namespace std;
typedef long long ll;
typedef pair<ll, ll> P;

ll gcd(ll a, ll b) {
	if(b == 0) return a;
	return gcd(b, a % b);
}

ll solve(vector<ll> c, int m, int n) {
	ll res = 0;
	for(ll i = 1; i < (1 << m); i++) {
		ll num = 0;
		for(ll j = i; j != 0; j >>= 1) num += j & 1;
		ll lcm = 1;
		for(ll j = 0; j < m; ++j) {
			if((i >> j) & 1) {
				lcm = lcm / gcd(lcm, c[j]) * c[j];
				if(lcm > n) break;
			}
		}
		if(num % 2 == 0) res -= n / lcm;
		else res += n / lcm;
	}
	return res;
}

int main() {
	int n, l, h;
	cin>>n>>l>>h;
	vector<ll> c(n);
	REP(i, n) cin>>c[i];
	ll res = 0;
	ll a = solve(c, n, h);
	ll z = 0;
	if(l > 1) z = solve(c, n, l - 1);
	REP(i, n) {
		vector<ll> b;
		REP(j, n) {
			if(j == i) continue;
			else b.push_back(c[j]);
		}
		res += a - solve(b, n - 1, h);
		if(l > 1) res -= z - solve(b, n - 1, l - 1);
	}
	cout<<res<<endl;
}