Thief in a Shop

n个物品每个价值ai,要求选k个,可以重复。问能取到哪几个价值?

1 ≤ n, k ≤ 1000,1 ≤ ai ≤ 1000

题解

将选一个物品能取到的价值的01生成函数k次方即可得到选k个物品得到的某个权值的方案数。

出题人卡NTT模数,998244353和1004535809都会被卡。然而469762049没被卡……

CO int N=1048576;
int a[N];
int rev[N],omg[N]; void NTT(int a[],int lim){
for(int i=0;i<lim;++i)
if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int i=1;i<lim;i<<=1)
for(int j=0;j<lim;j+=i<<1)
for(int k=0;k<i;++k){
int t=mul(omg[lim/(i<<1)*k],a[j+i+k]);
a[j+i+k]=add(a[j+k],mod-t),a[j+k]=add(a[j+k],t);
}
} int main(){
int n=read<int>(),m=read<int>();
while(n--) a[read<int>()]=1;
int len=ceil(log2(m*1000+1)),lim=1<<len;
for(int i=0;i<lim;++i) rev[i]=rev[i>>1]>>1|(i&1)<<(len-1);
omg[0]=1,omg[1]=fpow(3,(mod-1)/lim);
for(int i=2;i<lim;++i) omg[i]=mul(omg[i-1],omg[1]);
NTT(a,lim);
for(int i=0;i<lim;++i) a[i]=fpow(a[i],m);
omg[0]=1,omg[1]=fpow(omg[1],mod-2);
for(int i=2;i<lim;++i) omg[i]=mul(omg[i-1],omg[1]);
NTT(a,lim);
int ilim=fpow(lim,mod-2);
for(int i=0;i<lim;++i) a[i]=mul(a[i],ilim);
for(int i=0;i<lim;++i)
if(a[i]) printf("%d ",i);
return 0;
}

这题还有背包DP做法。将权值排序,对于i>1,把ai-=a1。然后做一个最短路径DP,只要路径长度≤k就可以凑出来。

之所以要减去a1,是因为最短路径长度<k的结果相当于用了(k-长度)个a1

int n,k,a[1005],dp[1000005];
int main(){
n=read(),k=read();
for(int i=0;i<n;i++)
a[i]=read();
sort(a,a+n);
for(int i=1;i<n;i++)
a[i]-=a[0];
for(int i=1;i<=k*a[n-1];i++)
dp[i]=3000;
for(int i=1;i<n;i++)
for(int j=a[i];j<=a[i]*k;j++)
dp[j]=min(dp[j],dp[j-a[i]]+1);
for(int i=0;i<=k*a[n-1];i++)
if(dp[i]<=k)printf("%d ",k*a[0]+i);
return 0;
}

Lightsabers (hard)

有\(n\)个有颜色的球,颜色编号为\(1\dots m\)中的一个。现在让你从中拿\(k\)个球,问拿到的球的颜色所构成的可重集合有多少种不同的可能。

注意同种颜色球是等价的,但是两个颜色为\(x\)的球不等价于一个。

\(1\leq n\leq 2\times 10^5,\quad 1\leq m,k\leq n。\)

题解

此题把每种颜色能选的数量的01生成函数乘起来就好了。

但是有个问题,那就是如果对每个多项式都做长度为\(k\)的FFT的话,是会TLE的。

所以需要用到启发式合并,每次选两个长度最小的多项式进行合并。可以用堆来维护,时间复杂度\(O(n \log^2 n)\)。

可以使用STL的make_heappush_heappop_heap系列函数,比priority_queue更快。

co double pi=acos(-1);
struct node {
double x,y;
node(){}
node(double x,double y):x(x),y(y){}
};
il node operator+(co node&a,co node&b){
return (node){a.x+b.x,a.y+b.y};
}
il node operator-(co node&a,co node&b){
return (node){a.x-b.x,a.y-b.y};
}
il node operator*(co node&a,co node&b){
return (node){a.x*b.x-a.y*b.y,a.x*b.y+a.y*b.x};
} co int N=1<<18,mod=1009;
int n,m,k,num[N]; node w[N],A[N],B[N];
int rev[N]; void trans(node a[],int lim){
for(int i=0;i<lim;++i)
if(i<rev[i]) swap(a[i],a[rev[i]]);
for(int step=1;step<lim;step<<=1){
int quot=lim/(step<<1);
for(int i=0;i<lim;i+=step<<1){
int j=i+step;
for(int k=0;k<step;++k){
node t=w[quot*k]*a[j+k];
a[j+k]=a[i+k]-t,a[i+k]=a[i+k]+t;
}
}
}
}
void mul_to(co vector<int>&a,co vector<int>&b,vector<int>&c){
for(int i=0;i<a.size();++i) A[i]=(node){a[i],0};
for(int i=0;i<b.size();++i) B[i]=(node){b[i],0};
int len=ceil(log2(a.size()+b.size()-1)),lim=1<<len;
for(int i=0;i<lim;++i){
rev[i]=rev[i>>1]>>1|(i&1)<<(len-1);
w[i]=(node){cos(i*2*pi/lim),sin(i*2*pi/lim)};
}
for(int i=a.size();i<lim;++i) A[i]=(node){0,0};
for(int i=b.size();i<lim;++i) B[i]=(node){0,0};
trans(A,lim),trans(B,lim);
for(int i=0;i<lim;++i){
A[i]=A[i]*B[i];
w[i].y=-w[i].y;
}
trans(A,lim);
c.resize(a.size()+b.size()-1);
for(int i=0;i<c.size();++i)
c[i]=(long long)round(A[i].x/lim)%mod;
} vector<int> col[N<<1];
int tot; il bool cmp(int a,int b){
return col[a].size()>col[b].size();
}
int heap[N],siz; int main(){
read(n),read(m),read(k);
for(int i=1;i<=n;++i) ++num[read<int>()];
for(int i=1;i<=m;++i){
if(!num[i]) continue;
++tot,col[tot].resize(num[i]+1);
for(int j=0;j<=num[i];++j) col[tot][j]=1;
heap[++siz]=tot;
}
make_heap(heap+1,heap+siz+1,cmp);
while(siz>=2){
int x=heap[1];
pop_heap(heap+1,heap+siz+1,cmp),--siz;
int y=heap[1];
pop_heap(heap+1,heap+siz+1,cmp),--siz;
mul_to(col[x],col[y],col[++tot]);
heap[++siz]=tot,push_heap(heap+1,heap+siz+1,cmp);
}
printf("%d\n",col[tot][k]);
return 0;
}

CF632E Thief in a Shop 和 CF958F3 Lightsabers (hard)的更多相关文章

  1. [CF632E]Thief in a Shop

    题目大意:有一个小偷,拿$k$个东西,有$n$种产品,每种产品都有无限多个.对于每个第$i$ 种产品,它的价值是$A_i$.可能偷走的物品价值之和. 题解:对于所有的物品构造生成函数$F(x)=\su ...

  2. codeforces 632+ E. Thief in a Shop

    E. Thief in a Shop time limit per test 5 seconds memory limit per test 512 megabytes input standard ...

  3. codeforces 632E. Thief in a Shop fft

    题目链接 E. Thief in a Shop time limit per test 5 seconds memory limit per test 512 megabytes input stan ...

  4. C - Thief in a Shop - dp完全背包-FFT生成函数

    C - Thief in a Shop 思路 :严格的控制好k的这个数量,这就是个裸完全背包问题.(复杂度最极端会到1e9) 他们随意原来随意组合的方案,与他们都减去 最小的 一个 a[ i ] 组合 ...

  5. codeforces Educational Codeforces Round 9 E - Thief in a Shop

    E - Thief in a Shop 题目大意:给你n ( n <= 1000)个物品每个物品的价值为ai (ai <= 1000),你只能恰好取k个物品,问你能组成哪些价值. 思路:我 ...

  6. Educational Codeforces Round 9 E. Thief in a Shop dp fft

    E. Thief in a Shop 题目连接: http://www.codeforces.com/contest/632/problem/E Description A thief made hi ...

  7. Educational Codeforces Round 9 E. Thief in a Shop NTT

    E. Thief in a Shop   A thief made his way to a shop. As usual he has his lucky knapsack with him. Th ...

  8. CF632E: Thief in a Shop(快速幂+NTT)(存疑)

    A thief made his way to a shop. As usual he has his lucky knapsack with him. The knapsack can contai ...

  9. Codeforces632E Thief in a Shop(NTT + 快速幂)

    题目 Source http://codeforces.com/contest/632/problem/E Description A thief made his way to a shop. As ...

随机推荐

  1. Kafka Consumer Lag Monitoring

    Sematext Monitoring  是最全面的Kafka监视解决方案之一,可捕获约200个Kafka指标,包括Kafka Broker,Producer和Consumer指标.尽管其中许多指标很 ...

  2. GIT 安装和使用

    目录 GIT 安装和使用 一.GIT 介绍 二.GIT 安装 三.GIT 使用 1. 配置 2. 创建版本库 3. 远程仓库 4. 分支管理 5.标签管理 6. 自定义 GIT 安装和使用 一.GIT ...

  3. 62 网络编程(三)——UDP编程

    UDP编程标准步骤 服务器端 使用DatagramSocket创建服务端:DatagramSocket server = new DatagramSocket(port);//参数为自定义端口号 准备 ...

  4. Git新建分支,分支合并,版本回退详解

    一.git基本命令 git拉取仓库代码 #拉取master代码 git clone git仓库地址 #拉取分支代码 git clone -b 分支名称 git仓库地址 2.git添加代码到本地仓库 g ...

  5. 使用Dbvisualizer 连接 Elasticsearch

    Dbvisualizer 安装 从网上下载该软件,并破解激活 下载地址:http://www.ddooo.com/softdown/142713.htm 1.下载解压,得到dbvisualizer p ...

  6. java中什么是抽象类(abstract)

    一.什么是抽象类 由abstract修饰的方法叫抽象方法:由abstract修饰的类叫抽象类.抽象的类无法进行实例化,因为他不是具体存在的类,或者说这样的类还不够完善,不能直接使用new关键字调用其构 ...

  7. C# Winform 文本框默认提示信息

    private string Notes = "提示文本"; private void textBox1_Leave(object sender, EventArgs e) { / ...

  8. javascript query string

    function getQueryVariable(variable) { var query = window.location.search.substring(1); var vars = qu ...

  9. Redis 获取和设置密码

    1.config get reuqirepass //获取当前密码 2.config set requirepass "password"//设置当前密码,双引号里面为密码

  10. 像Java一样管理对象:T&形式仅仅用在参数传递

    类的对象为了关联/包含一个T类型的instance,若成员变量包括T*/ T&, 这种设计叫做“aggregation”(聚合):而若采用T 形式,则称为"composition&q ...