luogu原题:蚯蚓
本题卡时间
时间复杂度:O(n+m)
开O2的AC代码(无读入优化)
Update(10月4日):
这题大家可以百度哈夫曼树
哈夫曼树有两种做法,一种是二叉堆O(nlogn)
我们把原来的二叉堆分成三份
发现它具有单调性
所以另一种是单调队列O(n)
由于我的不负责任
用O2水了过去
下面贴上我用读入优化,输出优化,inline优化的无O2的代码
完整AC代码
#include<cstdio>
#include<algorithm>
#define M 7000010
#define INF 2147483647
using namespace std;
int qu[3][M],h[3],g[3],w,n,m,q,u,v,t,num,tmp[20];
inline int rd(){
char ch;
int x=0,y=1;
while((ch=getchar())<'0'||ch>'9')if(ch=='-')y=-1;
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return x*y;
}
inline void wt(int x){
if(!x){
putchar('0');
return;
}
if(x<0)putchar('-'),x=-x;
tmp[0]=0;
while(x)tmp[++tmp[0]]=x%10,x/=10;
while(tmp[0])putchar(tmp[tmp[0]--]+'0');
printf(" ");
}
inline bool cmp(int p,int q){
return p>q;
}
inline int MAXN(){
int ma=-INF,k;
for(int i=0;i<=2;i++)if(g[i]>=h[i]&&qu[i][h[i]]>ma)k=i,ma=qu[i][h[i]];
h[k]++;
return qu[k][h[k]-1]+w;
}
int main(){
freopen("data.txt","r",stdin);
n=rd(),m=rd(),q=rd(),u=rd(),v=rd(),t=rd();
for(int i=1;i<=n;i++)qu[0][i]=rd();
sort(qu[0]+1,qu[0]+n+1,cmp);
h[0]=h[1]=h[2]=1;
g[0]=n;
for(int i=1;i<=m;i++){
num=MAXN();
w+=q;
if(i%t==0)wt(num);
qu[1][++g[1]]=(long long)num*u/v-w;
qu[2][++g[2]]=num-(long long)num*u/v-w;
}
printf("\n");
for(int i=1;i<=m+n;i++){
num=MAXN();
if(i%t==0)wt(num);
}
}
旧程序:
#pra\
gma GCC optimize("O2")
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define M 7000010
#define INF 2147483647
int qu[3][M],h[3],g[3],w,n,m,q,u,v,t,num;
bool cmp(int p,int q){
return p>q;
}
int MAXN(){
int ma=-INF,k;
for(int i=0;i<=2;i++){
if(g[i]>=h[i]&&qu[i][h[i]]>ma)k=i,ma=qu[i][h[i]];
}
h[k]++;
return qu[k][h[k]-1]+w;
}
int main(){
freopen("data.txt","r",stdin);
scanf("%d%d%d%d%d%d",&n,&m,&q,&u,&v,&t);
for(int i=1;i<=n;i++)scanf("%d",&qu[0][i]);
sort(qu[0]+1,qu[0]+n+1,cmp);
h[0]=h[1]=h[2]=1;
g[0]=n;
for(int i=1;i<=m;i++){
num=MAXN();
w+=q;
if(i%t==0)printf("%d ",num);
qu[1][++g[1]]=(long long)num*u/v-w;
qu[2][++g[2]]=num-(long long)num*u/v-w;
}
printf("\n");
for(int i=1;i<=m+n;i++){
num=MAXN();
if(i%t==0)printf("%d ",num);
}
}