代码源每日一题Div2
501. RSA
原题链接:RSA
思路:分别找这两个数的所有因数,如果两个数因数只有1和本身则两者均为质数,如果两个数有相同的因数或者两个数的因数出现了平方数则两数之积为大于1的整数的平方的整数倍。
代码:
#include <bits/stdc++.h>
using namespace std;
long long a, b, sum;
map<int, bool> mp;
bool check(int x){
int tmp = sqrt(x);
return tmp * tmp == x;
}
int main(){
cin >>a >> b;
if(a == b){
cout << "no credit";
return 0;
}
else{
mp[1] = true;
for(int i = 2; i <= sqrt(a); i++){
if(a % i == 0){
mp[i] = true;
mp[a / i] = true;
if(check(i) || check(a / i) || i == a / i){
cout << "no credit";
return 0;
}
}
}
for(int i = 2; i < sqrt(b); i++){
if(b % i == 0){
if(mp.count(i) || mp.count(b / i)){
cout << "no credit";
return 0;
}
else{
mp[i] = true;
mp[b / i] = true;
if(check(i) || check(b / i)){
cout << "no credit";
return 0;
}
}
}
}
for(auto x : mp){
sum++;
}
if(sum == 1) cout << "full credit";
else cout << "partial credit";
}
return 0;
}
502. 数组操作
原题链接:数组操作
思路:根据题意,每次的最优操作为将数组右边所有相等的数按照题目所给的操作要求赋值给左边的数,模拟次操作并记录操作次数即可。
代码:
#include <bits/stdc++.h>
using namespace std;
int t, n, a[200005];
int main(){
scanf("%d", &t);
for(int i = 0; i < t; i++){
scanf("%d", &n);
int ans = 0;
for(int j = 1; j <= n; j++) scanf("%d", &a[j]);
bool flag = true;
int sum = 1;
for(int j = n; j > 1; j--){
if(a[j] == a[j - 1]) sum++;
else{
ans++;
for(int k = 1; k <= sum; k++){
if(j - k >= 1) a[j - k] = a[j];
}
j = j - sum + 1, sum *= 2;
}
}
printf("%d\n", ans);
}
return 0;
}
503. A-B 数对
原题链接:A-B 数对
思路:要找到所有的A - B = C,即找到所有的A = B + C,因此用map存储所有的数并找到B + C是否存在于map中即可。又因为不同位置的数字一样的数对算不同的数对,因此还需要记录同一个数字出现的次数,表示每个位置都有一组不同的数对。
代码:
#include <bits/stdc++.h>
using namespace std;
long long n, c, a[200005], ans;
map<long long, bool> mp1;
map<long long, long long> mp2;
int main(){
scanf("%lld %lld", &n, &c);
for(int i = 1; i <= n; i++){
scanf("%lld", &a[i]);
if(!mp1.count(a[i])) mp1[a[i]] = true, mp2[a[i]] = 1;
else mp2[a[i]]++;
}
for(int i = 1; i <= n; i++){
if(mp1.count(a[i] + c)) ans += mp2[a[i] + c];
}
cout << ans;
return 0;
}
504. 数位计算
原题链接:数位计算
思路:观察题目规律,可以发现f(1) + … + f(N)为不同位数的的所有数构成的等差数列之和,用求和公式计算即可。由于数可能很大,因此我使用了__int128来储存数据。
代码:
#include <bits/stdc++.h>
using namespace std;
__int128 num, ans, flag = 99;
inline __int128 read(){
__int128 x = 0, f = 1;
char ch = getchar();
while(ch < '0' || ch > '9'){
if(ch == '-')
f = -1;
ch = getchar();
}
while(ch >= '0' && ch <= '9'){
x = x * 10 + ch - '0';
ch = getchar();
}
return x * f;
}
inline void print(__int128 x){
if(x < 0){
putchar('-');
x = -x;
}
if(x > 9)
print(x / 10);
putchar(x % 10 + '0');
}
int main(){
num = read();
if(num >= 10) ans = ans + 45;
else{
ans = (num + 1) * num / 2;
print(ans);
return 0;
}
while(num > flag){
__int128 tmp = flag - (flag / 10);
ans = (ans + (1 + tmp) * tmp / 2) % 998244353;
flag = flag * 10 + 9;
}
__int128 tmp = num - flag / 10;
ans = (ans + (1 + tmp) * tmp / 2) % 998244353;
print(ans);
return 0;
}
505. 新国王游戏
原题链接:新国王游戏
思路:题目要求找到值最大的排列顺序,那么我们可以直接对数组排序。在比较交换数组中相邻元素后总的最大值的大小时,由于其他位置的元素不受影响,因此我们可以得到这样的式子:
b1 * a2 * a3 * … * an + b2 * a3 * … * an > b2 * a1 * a3 * … * an + b1 * a3 * … * an
化简得:
b1 * (a2 - 1) > b2 * (a1 - 1)
根据这个式子用sort函数排序即可。
代码:
#include <bits/stdc++.h>
using namespace std;
long long n, ans, tmp, sum[1000005];
struct man{
long long a, b;
} m[1000005];
bool cmp(man x, man y){
return x.b * (y.a - 1) > y.b * (x.a - 1);
}
int main(){
scanf("%lld", &n);
for(int i = 1; i <= n; i++){
scanf("%lld %lld", &m[i].a, &m[i].b);
}
sort(m + 1, m + 1 + n, cmp);
sum[n + 1] = 1;
for(int i = n; i > 0; i--){
sum[i] = (sum[i + 1] * m[i].a) % 1000000007;
}
for(int i = 1; i <= n; i++){
tmp = (m[i].b * sum[i + 1]) % 1000000007;
ans = (ans + tmp) % 1000000007;
}
printf("%lld", ans);
return 0;
}
506. 完美数
原题链接:完美数
思路:由于完美数的每一位都是给定的a或b,因此只要对a或b的出现次数进行遍历比找到满足所有a和b的和为好数即可。此时再用a和b排列组合即可求出完美数的数量。难点在于如何求数字极大的组合数取模后的数。
为了计算 a / b mod p的数值,此时需要引入逆元的概念。设inv(b)为b在mod p情况下的逆元,则a / b mod p = a * inv(b) mod p,运用逆元可以将除法转化为乘法,使得取模操作可以正常使用。下面是计算逆元的方法:
long long inv(long long x, long long p){
long long res = 1, y = p - 2;
while(y){
if(y & 1) res = res * x % p;
x = x * x % p;
y >>= 1;
}
return res;
}
inv(x, p)表示x在mod p情况下的逆元。
整体代码:
#include <bits/stdc++.h>
using namespace std;
long long a, b, m, ans, fact[1000005];
const int mod = 1e9 + 7;
bool check(long long x){
long long tmp = x;
while(tmp){
if(tmp % 10 != a && tmp % 10 != b){
return false;
}
tmp /= 10;
}
return true;
}
long long inv(long long x, long long p){
long long res = 1, y = p - 2;
while(y){
if(y & 1) res = res * x % p;
x = x * x % p;
y >>= 1;
}
return res;
}
long long C(long long x, long long y){
return (fact[x] * inv(fact[y], mod) % mod) * inv(fact[x - y], mod) % mod;
}
int main(){
cin >> a >> b >> m;
fact[0] = 1;
for(int i = 1; i <= m; i++){
fact[i] = (fact[i - 1] * i) % mod;
}
for(int i = 0; i <= m; i++){
if(check(i * a + (m - i) * b)) ans = (ans + C(m, i)) % mod;
}
cout << ans;
return 0;
}
507. Lusir的游戏
原题链接:Lusir的游戏
思路:二分查找符合条件的最小能量值即可。
代码:
#include <bits/stdc++.h>
using namespace std;
int n, h[100005];
bool check(long long e){
for(int i = 1; i <= n; i++){
e = 2 * e - h[i];
if(e < 0) return false;
if(e > 100005) return true;
}
return true;
}
int main(){
cin >> n;
for(int i = 1; i <= n; i++){
cin >> h[i];
}
int st = 1, ed = 100005;
while(st <= ed){
int mid = (st + ed) / 2;
if(check(mid)) ed = mid - 1;
else st = mid + 1;
}
cout << st;
return 0;
}
601. BFS练习1
原题链接:BFS练习1
思路:用bfs把a变为b1,b2,…,bn的最少步数全都求出来,然后询问时直接对照预处理的数组输出即可。
代码:
#include <bits/stdc++.h>
using namespace std;
int a, q, b, step[1000005];
bool vis[1000005];
void bfs(int x){
queue<int> que;
que.push(x);
vis[x] = true;
while(!que.empty()){
int num = que.front();
que.pop();
for(int i = 0; i < 4; i++){
int tmp = num;
if(i == 0){
tmp += 1;
if(!vis[tmp] && tmp <= 100000){
vis[tmp] = true;
que.push(tmp);
step[tmp] = step[num] + 1;
}
}
else if(i == 1){
tmp *= 2;
if(!vis[tmp] && tmp <= 100000){
vis[tmp] = true;
que.push(tmp);
step[tmp] = step[num] + 1;
}
}
else if(i == 2){
tmp *= 3;
if(!vis[tmp] && tmp <= 100000){
vis[tmp] = true;
que.push(tmp);
step[tmp] = step[num] + 1;
}
}
else if(i == 3){
tmp -= 1;
if(!vis[tmp] && tmp >= 1){
vis[tmp] = true;
que.push(tmp);
step[tmp] = step[num] + 1;
}
}
}
}
return;
}
int main(){
scanf("%d %d", &a, &q);
bfs(a);
for(int i = 0; i < q; i++){
scanf("%d", &b);
printf("%d ", step[b]);
}
return 0;
}