Codeforces Gym 101190 NEERC 16 .L List of Primes(递归)

时间:2023-03-08 21:16:40
ls特别喜欢素数,他总是喜欢把素数集合的所有子集写下来,并按照一定的顺序和格式。对于每一个子集,集合内
的元素在写下来时是按照升序排序的,对于若干个集合,则以集合元素之和作为第一关键字,集合的字典序作为第
二关键字(先比较集合第一个元素,再比较第二个元素,以此类推),这个序列的开始如下:[2], [3], [2, 3], 
[5], [2, 5], [7], [3, 5], [2, 7], [2, 3, 5], [3, 7], [11], [2, 3, 7], [5, 7], [2, 11], [13], [2, 5, 
7]......注意:每个逗号的后面均有一个空格。现在ls想询问该序列位于区间[a,b]的子串是什么。

Input

输入仅一行包含两个数:a,b(1<=a<=b<=1e18,b-a<=100000)。

Output

输出序列中位于区间[a,b]的子串,前置或后置空格也应输出。

Sample Input

1 35

Sample Output

[2], [3], [2, 3], [5], [2, 5], [7],

题意:把素数集合按照和的大小排序,和相同的按照字典序排序,现在把这个排序后的序列看成一个字符串,问某个区间的字符串是什么。

思路:注意到素数不会太多,只有300来个,素数的和也不会太大,不会超过3000,所以我们递归去找即可。反正就是像数位DP那样搞就行了。。。

用记忆化递推缩小范围,然后用回溯暴力出ans。

#include<bits/stdc++.h>
#define ll long long
#define P pair<ll,ll>
using namespace std;
const int maxn=1e6;
int isprime[maxn],p[maxn];
vector<int>primes;
vector<int>vec;
map<P,P>M;
ll a,b,cur,prefix_len;
int getL(int x){
int res=;
while(x>){ res++; x/=;}
return res+;
}
inline P get(P p1,P p2,ll len){
return P(p1.first+p2.first,p1.second+p2.second+len*p1.first);
}
P calc(int x,int sum)
{
P p(x,sum);
if(sum<) return P(,);
if(M.count(p)) return M[p];
if(sum==) return M[p]=P(,); //ct,len
if(primes[x]>sum) return P(,);
return M[p]=get(calc(x+,sum-primes[x]),calc(x+,sum),getL(primes[x]));
}
void ptchar(char c){
cur++; if(cur>=a&&cur<=b) putchar(c);
}
void print(int x){
vector<int>v;
while(x>){ v.push_back(x%); x/=;}
for(int i=v.size()-;i>=;i--) ptchar(char(v[i]+''));
}
void print(int x,int sum)
{
if(sum<||cur>=b) return ;
if(sum==){
ptchar('[');
for(int i=;i<vec.size();i++){
print(vec[i]);
if(i==vec.size()-) ptchar(']');
ptchar(',');
ptchar(' ');
} return ;
}
if(primes[x]>sum) return ;
vec.push_back(primes[x]);
prefix_len+=getL(primes[x]);
ll len=prefix_len*calc(x+,sum-primes[x]).first+calc(x+,sum-primes[x]).second;
if(len+cur>=a) print(x+,sum-primes[x]);
else cur+=len; vec.pop_back();
prefix_len-=getL(primes[x]);
len=prefix_len*calc(x+,sum).first+calc(x+,sum).second;
if(len+cur>=a) print(x+,sum);
else cur+=len;
}
int main()
{
freopen("list.in","r",stdin);
freopen("list.out","w",stdout);
isprime[]=false;
fill(isprime,isprime+maxn,true);
for(int i=;i<maxn;i++){
if(isprime[i]){
primes.push_back(i);
for(int j=i+i;j<maxn;j+=i) isprime[j]=false;
}
}
scanf("%I64d%I64d",&a,&b);
for(int i=;i<&&cur<b;i++){ //试探过,i最大到2096,这个范围的素数也就300来个,所以记忆化
ll len=calc(,i).second;
if(cur+len>=a) print(,i);
else cur+=len;
}
puts("");
return ;
}