题解 | #牛牛排队#

牛牛排队

http://www.nowcoder.com/practice/937cafd766f64553924ea96c0308ca48

描述

这是一篇面对初级coder的题解。
知识点:数学 排序 快速幂
难度:二星


题解

题目:
一共n个人,已知之前站在他们左部分和站在他们右部分的人的人数差的绝对值,求有多少种不同的站法?(结果需要对1e9+7取模)
分析:
本题的关键在于判断队列是否存在,不难发现,站在他左部分和站在他右部分的人的人数差的绝对值是有一定规律的
如下图
图片说明
之后只需要根据n是奇数和偶数,判断数组是否符合规范即可:
若数组符合规范,则最终结果为
但是由于数字较大,同时模的值也大需要用到

中的快速幂和快速乘算法

方法一:字典map

采用一个map存储

class Solution {
public:
      const long long mod=1000000007;
      long long power(long long base, long long exponent) {
        if(exponent==0)
            return 1;
        long long answer = 1;
        while(exponent){
            if(exponent&1)        //如果n的当前末位为1
                answer =muti(answer,base);  //ans乘上当前的a
            base =muti(base,base);        //a自乘
            exponent >>= 1;       //n往右移一位
        }
        return answer;
    }
    long long muti(long long a, long long b) {//防止越界 乘法 每次只乘2字节
        long long ans =0;
        while(b){
            ans=(ans+a*(b&0xFFFFLL))%mod;//相乘累加
            b>>=16;//乘数右移2字节
            a=(a<<16)%mod;//被乘数左移2字节
        }
        return ans;
    }
    int solve(int n, vector<int>& a) {
        vector<int> map(n+1,0);//记录每个数字出现的次数
        for(int i=0;i<n;i++)
            map[a[i]]++;//记录
        int begin;
        if(n%2==1){//分奇数偶数讨论
            begin=2;//奇数 看2 4 6 8 是否为2
            if(map[0]!=1)//看 0是否为1
                return 0;
        }
        else
            begin=1;//偶数 看1 3 5 7 是否为2
        for(int i=begin;i<n+1;i+=2){
            if(map[i]!=2)
                return 0;
        }
        return (int)power(2,n/2);//若合法 返回2^(n/2)
        // write code here
    }
};
复杂度分析:
  • 时间复杂度:,统计数字遍历一遍,检验遍历一半 快速幂
  • 空间复杂度:,map数组的长度

运行测试:
运行时间 3ms 占用内存 304KB

方法二:排序

先将a数组排序,再统计是否符合要求

class Solution {
public:
      const long long mod=1000000007;
      long long power(long long base, long long exponent) {
        if(exponent==0)
            return 1;
        long long answer = 1;
        while(exponent){
            if(exponent&1)        //如果n的当前末位为1
                answer =muti(answer,base);  //ans乘上当前的a
            base =muti(base,base);        //a自乘
            exponent >>= 1;       //n往右移一位
        }
        return answer;
    }
    long long muti(long long a, long long b) {//防止越界 乘法 每次只乘2字节
        long long ans =0;
        while(b){
            ans=(ans+a*(b&0xFFFFLL))%mod;//相乘累加
            b>>=16;//乘数右移2字节
            a=(a<<16)%mod;//被乘数左移2字节
        }
        return ans;
    }
    int solve(int n, vector<int>& a) {
        sort(a.begin(),a.end());
        int begin;
        int data;
        if(n%2==1){//分奇数偶数讨论
            begin=1;
            data=2;//奇数 从1开始是否为2 4 6 8
            if(a[0]!=0)//看 0是否为1
                return 0;
        }
        else{
            begin=0;//偶数 从0开始是否为1 3 5 7
            data=1;
        }
        for(int i=begin;i<n;i+=2)
        {
            if(a[i]!=data||a[i+1]!=data)//检验
                 return 0;
            data+=2;
        }
        return (int)power(2,n/2);//若合法 返回2^(n/2)
        // write code here
    }
};
复杂度分析:
时间复杂度::快速排序sort函数,检查数组是否合法为,快速幂
空间复杂度:主要是快速排序sort函数

运行测试:
运行时间 3ms 占用内存 428KB

总结

先用数学分析,再做计算
排序 字典 基础知识灵活应用
阿Q的题解 文章被收录于专栏

阿Q秋招刷过的题

全部评论

相关推荐

不愿透露姓名的神秘牛友
昨天 17:16
科大讯飞 算法工程师 28.0k*14.0, 百分之三十是绩效,惯例只发0.9
点赞 评论 收藏
分享
10-28 15:45
门头沟学院 C++
西南山:海康威视之前不是大规模裁员吗
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务