好友
阅读权限10
听众
最后登录1970-1-1
|
本帖最后由 1736809 于 2024-5-4 17:23 编辑
编程小白,最近在学C语言,想着练一练,并发帖记录一下
49. 字母异位词分组
字母异位词 是由重新排列源单词的所有字母得到的一个新单词。 示例
输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
花了1个多小时,总算做出来了。
就是消耗内存很大,求大佬指点,提出改进的思路
[C] 纯文本查看 复制代码 struct pair{
char * s1;
char * s1_qsort;
};
typedef struct pair Pair;
int cmp(void *a,void *b)
{
return (*(char *)a - *(char *)b);
//return ( *(int*)a - *(int*)b );
}
int paircmp(void *a,void *b)
{
Pair * Pa=(Pair*)a;
Pair * Pb=(Pair*)b;
return strcmp(Pa->s1_qsort,Pb->s1_qsort);
}
char*** groupAnagrams(char** strs, int strsSize, int* returnSize, int** returnColumnSizes) {
*returnSize=0;
* returnColumnSizes=NULL;
char***res=NULL;
Pair * pairs=(Pair *)malloc(sizeof(Pair)*strsSize);
for(int i=0;i<strsSize;i++)
{
pairs[i].s1=strs[i];
pairs[i].s1_qsort=(char*)malloc(sizeof(char)*(strlen(strs[i])+1));
strcpy(pairs[i].s1_qsort,pairs[i].s1);
qsort(pairs[i].s1_qsort,strlen(strs[i]),sizeof(char),cmp);
}
qsort(pairs,strsSize,sizeof(Pair),paircmp);
int group=-1;//记录最后组坐标
int index=0;//记录组内坐标
for(int i=0;i<strsSize;i++)
{
if(i==0||strcmp(pairs[i-1].s1_qsort,pairs[i].s1_qsort)!=0)//no equal
{
group++;
index=0;
*returnSize=group+1;//group++
*returnColumnSizes=(int*)realloc(*returnColumnSizes,sizeof(int)*(group+1));
(*returnColumnSizes)[group]=index+1;
res=(char***)realloc(res,sizeof(char**)*(group+1));
res[group]=(char**)malloc(sizeof(char*)*(index+1));
}
else//equal
{
index++;
(*returnColumnSizes)[group]=index+1;
res[group]=(char**)realloc(res[group],sizeof(char*)*(index+1));
}
res[group][index]=(char*)malloc(sizeof(char)*(strlen(pairs[i].s1)+1));
strcpy(res[group][index],pairs[i].s1);
}
return res;
}
2024年5月4日更新
我把realloc相关函数删掉了,改成先计算组数和每组的个数,内存下降了很多,特地分享下成功的喜悦
优化后的代码:
[C] 纯文本查看 复制代码 struct pair{
char * s1;
char * s1_qsort;
};
typedef struct pair Pair;
int cmp(void *a,void *b)
{
return (*(char *)a - *(char *)b);
//return ( *(int*)a - *(int*)b );
}
int paircmp(void *a,void *b)
{
Pair * Pa=(Pair*)a;
Pair * Pb=(Pair*)b;
return strcmp(Pa->s1_qsort,Pb->s1_qsort);
}
char*** groupAnagrams(char** strs, int strsSize, int* returnSize, int** returnColumnSizes) {
int NofG[10000]={0};//记录每组总个数
int Gs=0,Gc=0;
*returnSize=0;
* returnColumnSizes=NULL;
char***res=NULL;
Pair * pairs=(Pair *)malloc(sizeof(Pair)*strsSize);
for(int i=0;i<strsSize;i++)
{
pairs[i].s1=strs[i];
pairs[i].s1_qsort=(char*)malloc(sizeof(char)*(strlen(strs[i])+1));
strcpy(pairs[i].s1_qsort,pairs[i].s1);
qsort(pairs[i].s1_qsort,strlen(strs[i]),sizeof(char),cmp);
}
qsort(pairs,strsSize,sizeof(Pair),paircmp);
for(int i=0;i<strsSize;i++)
{
if(i==0)//no equal
{
Gc=1;
}
else
{
if(strcmp(pairs[i-1].s1_qsort,pairs[i].s1_qsort)!=0)//no equal
{
NofG[Gs]=Gc;
//printf("%d NofG[Gs]=%d\n",i,NofG[Gs]);
Gc=1;
Gs++;
}
else
{
Gc++;
}
}
}
NofG[Gs++]=Gc;
res=(char***)malloc(sizeof(char**)*Gs);
*returnColumnSizes=(int*)malloc(sizeof(int)*(Gs));
*returnSize=Gs;
for(int i=0;i<Gs;i++)
{
res[i]=(char**)malloc(sizeof(char*)*(NofG[i]));
(*returnColumnSizes)[i]=NofG[i];
}
int group=-1;//记录最后组坐标
int index=0;//记录组内坐标
for(int i=0;i<strsSize;i++)
{
if(i==0||strcmp(pairs[i-1].s1_qsort,pairs[i].s1_qsort)!=0)//no equal
{
group++;
index=0;
}
else//equal
{
index++;
}
res[group][index]=(char*)malloc(sizeof(char)*(strlen(pairs[i].s1)+1));
strcpy(res[group][index],pairs[i].s1);
}
return res;
} |
|