kkun的一道简单签到题
发布时间: 2017年6月11日 17:59 最后更新: 2017年6月11日 18:10 时间限制: 2000ms 内存限制: 512M描述
青大最优雅的程序员kkun一直友好(?)的致力于为学妹们出一道做法优雅而且非常简单的题。
例如著名的“一道非常简单的炉石题”,“一道非常简单的签到题”,“一道非常简单的数学题”,“一道非常简单的平衡树套替罪羊树套喜羊羊树套圣诞树套动态仙人掌”,“一道非常简单的签到题”(这个人是真的优雅,比那个菜凡不知道高到哪里去了!)
这次也不例外,他想从no cer那里拿到n个数,并找到其中最小的k个数。
但是no cer只是扔给了他一个公式:val[i]=k^(i+1)%1000000007(1<=i<=n),并让他自己一边玩去。
kkun顿时觉得这个问题变得索然无味,所以他让你来解决这个问题。
最终从大到小输出最小的k个数。
输入
第一行一个整数t代表数据组数
每组数据中
第一行两个整数n,k
n<=1e7
k<=1e4
k<=n
输出
从大到小输出k个数,每组数据占一行
样例输入1
1
3 2
样例输出1
8 4
一开始求每个数的val[i]时用的快速幂,然后超时了……
发现只要记录下前一个数然后一直乘以后一个数就行了,再用一个优先队列,始终维护k个最小的数
数据可能会超过int,改成long long 就过了
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
const long long c = 1000000007;
typedef long long ll;
priority_queue<ll> q;
int main()
{
ll n, k, T;
ll tmp, pre;
bool flag;
scanf("%d", &T);
while(T--)
{
scanf("%lld%lld", &n, &k);
pre = k;
for(int i=1;i<=n;i++)
{
tmp = (pre%c) * (k%c) % c;
pre = tmp;
if(i<=k)
q.push(tmp);
else if(q.top()>tmp)
{
q.pop();
q.push(tmp);
}
}
printf("%lld", q.top());
q.pop();
while(!q.empty())
{
printf(" %lld", q.top());
q.pop();
}
printf("\n");
}
return 0;
}