今天在洛谷刷到了一道简单的贪心算法题,在这里给大家分享一下,顺便提一下在做这道题的时候发现的一个巨坑(费尽心思才找出来的呜呜呜)
这是原题
P1056 排桌椅
仔细审题之后,发现这是一道贪心算法的简单应用,可以用两个数组分别存放每一行和每一列上交头接耳的人的对数,然后再分别开辟一个数组存放人数最多的前K和L个行和列,再进行一趟排序就可以AC了
上代码
#include <bits/stdc++.h>
using namespace std;
bool cmp(int a,int b){
return a<b;
}
int main()
{
int m,n,k,l,d;
int x,y,p,q;
int a1[1010],b1[1010];
int a[1010],b[1010];
//memset(a,0,sizeof(a));
//memset(b,0,sizeof(b));
//memset(a1,0,sizeof(a1));
//memset(b1,0,sizeof(b1));
cin>>m>>n>>k>>l>>d;
for(int i=0;i<d;i++){
cin>>x>>y>>p>>q;
if(x==p){
b[min(y,q)]++;
}
if(y==q){
a[min(x,p)]++;
}
}
int max,t;
for(int i=1;i<=k;i++){
max=0;t=0;
for(int j=1;j<=m;j++){
if(a[j]>max) {
max=a[j];
t=j;
}
}
a1[i]=t;
a[t]=0;
}
for(int i=1;i<=l;i++){
max=0;t=0;
for(int j=1;j<=n;j++){
if(b[j]>max) {
max=b[j];
t=j;
}
}
b1[i]=t;
b[t]=0;
}
sort(a1+1,a1+k+1,cmp);
sort(b1+1,b1+l+1,cmp);
for(int i=1;i<=k;i++)
cout<<a1[i]<<" ";
cout<<endl;
for(int i=1;i<=l;i++)
cout<<b1[i]<<" ";
return 0;
}
思路和代码都没有问题,但是我提交之后直接WA了,然后我就开始了漫长的寻bug过程
找到最后,这里就要回到文章开头说得那个巨坑了,仔细看我代码里那四个注释的内容,原来bug就在这里,我定义的数组都没有默认初始化,不是c++的数组默认初始化都为0吗,这谁顶得住啊
去网上查了一下,发现很多编译器在数组初始化时不一定都默认为0,那么就要我们自己去初始化了,这里给大家介绍两种重置为0的方法
1.在定义数组时,例如a[100]={0},注意:这种方法如果想把整形数组a都初始化为1就不能用了
比如int a[1000] = {1};//数组其实是被初始化成1,0,0,0,0
2.如果是在定义以后想全部赋为0,用memset()函数是比较快的。
int a[SIZE];
memset(a, 0, sizeof(a));
memset(a, 0, sizeof(int)*1000);//这里的1000是数组大小,需要多少替换下就可以了。
加上注释的内容之后,我就快乐地AC了,总的来说做完这道题还是有点收获,至少以后不会再进入今天这个坑了哈哈