其他分享
首页 > 其他分享> > 生存人数问题(Leetcode)

生存人数问题(Leetcode)

作者:互联网

题目描述

给定 N 个人的出生年份和死亡年份,第 i 个人的出生年份为 birth[i],死亡年份为 death[i],实现一个方法以计算生存人数最多的年份。

你可以假设所有人都出生于 1900 年至 2000 年(含 1900 和 2000 )之间。如果一个人在某一年的任意时期处于生存状态,那么他应该被纳入那一年的统计中。例如,生于 1908 年、死于 1909 年的人应当被列入 1908 年和 1909 年的计数。

如果有多个年份生存人数相同且均为最大值,输出其中最小的年份。
输入: birth = {1900, 1901, 1950} death = {1948, 1951, 2000} 输出: 1901

首先说下自己想出来的比较笨的方法

使用一个大小为101的一维辅助数组count[]存储1900年到2000年每一年的存活人数,比如(出生年为1900年,死亡年1948年)的人,他应该在辅助数组中count[0]~count[48]都加1,表示这些年里这个人都存在。

class Solution {
    public int maxAliveYear(int[] birth, int[] death) {
        int[] count = new int[101];
	    for(int i=0;i<birth.length;i++) {//遍历人员信息
		    for(int j=birth[i]-1900;j<=death[i]-1900;j++) {   //将一个人的各个生存年份填入到count数组中
			    count[j]++;
		    }	
	    }
        int max = count[0];
        int index=0;
        for(int i=1;i<count.length;i++){ //找到人数最多的年份
            if(count[i]>max){
                max=count[i];
                index=i;
            }
        }
        return index+1900;
    }
}

参考别人写的方法(前缀和)

利用前缀和的方式,减少一层for循环,使得时间复杂度降低。
只关心出生年份和死亡年份的人数变化,按题意,死亡当年也算作还在生存,意味着死亡后的第二年才有人数减少。

class Solution {
    public int maxAliveYear(int[] birth, int[] death) {
        int[] lives = new int[102];  //lives数组实质上反映的是某一年的人数变化
        for(int i=0;i<birth.length;i++){
            lives[birth[i]-1900]++;  //出生年份代表该年增加一人
            lives[death[i]-1900+1]--; //死亡年份代表次年减少一人
        }
        int[] preSum = new int[102]; 
        preSum[0] = lives[0]; 
        for(int i=1;i<preSum.length;i++){   
            preSum[i]=lives[i]+preSum[i-1];  
        }
        int max = preSum[0];
        int index = 0;
        for(int i=1;i<preSum.length;i++){
            if(preSum[i]>max){
                max = preSum[i];
                index = i;
            }
        }
        return index+1900;

    }
}

标签:count,年份,生存,int,birth,preSum,人数,Leetcode,1900
来源: https://blog.csdn.net/qq_38848965/article/details/115821511