此题初看题目,很快想到了暴力遍历的办法去求解,但这个程序的时间复杂度太高了(时间复杂度为n2),不禁让我后背发凉(我心想这应该不会如此简单)。
于是在暴力方式求解的程序提交后,果然出了问题。问题显示为最后一个测试点不通过,超时了。
//暴力遍历求解完整代码
#include <stdio.h>
typedef struct
{
int number;
int grade;
}pInfo;
int main()
{
int n;
scanf("%d",&n);
pInfo infos[100000];
int i,j;
for(i=0;i<n;i++)
{
scanf("%d %d",&infos[i].number,&infos[i].grade);
}
pInfo infosum[100000];
int count=1;
int flag;
infosum[0]=infos[0];
for(i=1;i<n;i++)
{
flag=0;
for(j=0;j<count;j++)
{
if(infos[i].number==infosum[j].number)
{
infosum[j].grade=infosum[j].grade+infos[i].grade;
flag=1;
break;
}
}
if(flag==0) infosum[count++]=infos[i];
}
int max=0;
for(i=1;i<count;i++)
{
if(infosum[i].grade>infosum[max].grade)
{
max=i;
}
}
printf("%d %d",infosum[max].number,infosum[max].grade);
return 0;
}
之后仔细思考如何优化程序过程,降低时间复杂度。
分析可知程序需要优化的地方就是对所有信息暴力遍历合并分数的那个两层的循环
//截取自上述代码
for(i=1;i<n;i++)
{
flag=0;
for(j=0;j<count;j++)
{
if(infos[i].number==infosum[j].number)
{
infosum[j].grade=infosum[j].grade+infos[i].grade;
flag=1;
break;
}
}
if(flag==0) infosum[count++]=infos[i];
}
问题主要在内循环,思考知这里主要实现了对同编号结构体的查找,这个遍历即查找的时间规模为n,于是自然而然的想到了使用折半查找(时间复杂度为log2n),这样整个程序的时间复杂度就从n2变为了nlog2n。
但要使用折半查找就得数据顺序存放且有序,这里编号显然是无序的,那么就还需要事先排序,自然使用快排(时间复杂度为nlog2n),排序本身不会增大程序的时间复杂度,到此很好了,于是似乎就可以开始了。
但我突然又发现这里还可以简化的地方,我都使用快速排序了,参赛选手的信息在排序后已经按照学校编号有序了(最为重要的是相同编号的必然连续放在一起!)。这里直接对排序后的数组进行一趟遍历,在遍历中每次标记第一个出现的不一样的编号,然后根据这个编号的总分进行累和,直到出现下一个不一样的编号再重新累和,这样不就可以了吗,并且就不涉及到两层循环且内循环是折半查找了。当然这种方案时间复杂度由于快排的存在依然是nlog2n,只是这种方案写起来简单点且程序局部是时间规模会小一点(因为只有排序后的一趟遍历即n,而折半查找的话遍历是双层的即nlog2n)。
于是有了如下的优化后的代码
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
typedef struct
{
int number;
int grade;
}pInfo;
bool cmp(pInfo p1,pInfo p2)
{
if(p1.number<p2.number) return true;
else return false;
}
int main()
{
int n;
scanf("%d",&n);
vector<pInfo> v,v1;
pInfo p;
int i;
for(i=0;i<n;i++)
{
scanf("%d %d",&p.number,&p.grade);
v.push_back(p);
}
sort(v.begin(),v.end(),cmp);
int j=0;
for(i=1;i<n;i++)
{
if(v[i].number==v[j].number)
{
v[j].grade=v[j].grade+v[i].grade;
}
else
{
v1.push_back(v[j]);
j=i;
}
}
v1.push_back(v[j]);
int max=0;
for(i=1;i<v1.size();i++)
{
if(v1[i].grade>v1[max].grade)
{
max=i;
}
}
printf("%d %d",v1[max].number,v1[max].grade);
return 0;
}