标签:取模 Johnny 双模 int res ll Grandmaster ans 集合
题意:
把 \(p^{k_1},p^{k_2},\cdots p^{k_n}\) 分成两个集合,使两个集合的总和的差的绝对值最小(是原数的差最小而不是取模后最小)。输出差的绝对值取模。
思路:
从大到小考虑每个数,如果 ans 为 0 则把当前数放入集合Ⅰ,即 ans 加上 \(p^i\);
如果 ans 大于 0,说明集合Ⅰ比集合Ⅱ的和大(实际上一定恰好多出来 \(p^j\) )。把数看成 p进制,后面的数一直放入集合Ⅱ直到 ans 等于0(说明用较小的数凑出了 \(p^j\))。
ans 一直要取模,判断取模后的数是不是真的为0要用双模法!
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e6 + 5, M1 = 1e9 + 7, M2 = 1e9 + 3;
int n, p, k[N];
int qmi(int m, int k, int p)
{
int res = 1 % p, t = m;
while (k)
{
if (k&1) res = (ll)res * t % p;
t = (ll)t * t % p;
k >>= 1;
}
return res;
}
signed main()
{
int T; cin >> T; while(T--)
{
scanf("%d%d", &n, &p);
for(int i = 1; i <= n; i++) scanf("%d", &k[i]);
sort(k + 1, k + 1 + n, greater<int>());
int ans1 = 0, ans2 = 0;
for(int i = 1; i <= n; i++)
{
if(ans1 || ans2) //不是0
ans1 = (ans1 - qmi(p, k[i], M1) + M1) % M1,
ans2 = (ans2 - qmi(p, k[i], M2) + M2) % M2;
else //是0
ans1 = (ans1 + qmi(p, k[i], M1)) % M1,
ans2 = (ans2 + qmi(p, k[i], M2)) % M2;
}
printf("%d\n", ans1);
}
return 0;
}
标签:取模,Johnny,双模,int,res,ll,Grandmaster,ans,集合 来源: https://www.cnblogs.com/wushansinger/p/15725798.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。