查看: 2668| 回复: 2
跳转到指定楼层
上一主题 下一主题
收起左侧

火车售票问题

全局:

注册一亩三分地论坛,查看更多干货!

您需要 登录 才可以下载或查看附件。没有帐号?注册账号

x
某次列车途经C个城市,城市编号依次为1到C,列车上共有S个座位,铁路局规定售出的车票只能是坐票,即车上所有的旅客都有座,售票系统是由计算机执行的,每一个售票申请包含三个参数,分别用O、D、N表示,O为起始站,D为目的地站,N为车票张数,售票系统对该售票申请作出受理或不受理的决定,只有在从O到D的区段内列车上都有N个或N个以上的空座位时该售票申请才被受理,请你写一个程序,实现这个自动售票系统。

提示:售票查询时间复杂度log(n) n为城市数目

上一篇:区间最大值问题
下一篇:求杨氏矩阵第k大的值
🔗
 楼主| wwwyhx 2011-6-11 12:10:18 | 只看该作者
全局:
线段树求解,建立线段树,线段树父节点记录所有孩子的最大值,需要一些技巧实现最大值的更新。
代码:
struct SEGMENT
{
        int nFrom;
        int nTo;
        int nMaxTickets;
        int nPassDown;

        SEGMENT(int nf = 0, int nt = 0) : nFrom(nf), nTo(nt)
        { nMaxTickets = 0; nPassDown = 0; }
};

class CTicketSystem
{
public:
        CTicketSystem(int nCities, int nSits);
        ~CTicketSystem() { delete []m_pTree; }

public:
        bool BuyTicket(int nBeg, int nEnd, int n);

private:
        void _inner_build(int nIndex, int nBeg, int nEnd);
        bool _inner_judge(int nIndex, int nBeg, int nEnd, int n, int nPassDownSum);
        void _inner_add(int nIndex, int nBeg, int nEnd, int n);

private:
        SEGMENT* m_pTree;
        int m_nCities;
        const int m_nSits;
};

inline CTicketSystem::CTicketSystem(int nCities, int nSits) : m_nCities(nCities), m_nSits(nSits)
{
        assert(nCities>0);
        assert(nSits>0);

        m_pTree = new SEGMENT[3*nCities];
        _inner_build(0, 0, m_nCities-1);
}

inline void CTicketSystem::_inner_build(int nIndex, int nBeg, int nEnd)
{
        m_pTree[nIndex].nFrom = nBeg;
        m_pTree[nIndex].nTo = nEnd;

        if (nBeg == nEnd) return;

        _inner_build(2*nIndex+1, nBeg, (nBeg+nEnd)/2);
        _inner_build(2*nIndex+2, (nBeg+nEnd)/2+1, nEnd);
}

inline bool CTicketSystem::BuyTicket(int nBeg, int nEnd, int n)
{
        assert(nBeg>=0 && nEnd>=0 && nEnd>=nBeg && n>0);

        if (_inner_judge(0, nBeg, nEnd, n, 0))
        {
                _inner_add(0, nBeg, nEnd, n);
                return true;
        }

        return false;
}

bool CTicketSystem::_inner_judge(int nIndex, int nBeg, int nEnd, int n, int nPassDownSum)
{
        nPassDownSum += m_pTree[nIndex].nPassDown;
        if (m_pTree[nIndex].nFrom == nBeg && m_pTree[nIndex].nTo == nEnd)
                return nPassDownSum + n <= m_nSits;

        int nMid = (m_pTree[nIndex].nFrom + m_pTree[nIndex].nTo)/2;
        if (nMid >= nEnd)
                return _inner_judge(2*nIndex+1, nBeg, nEnd, n, nPassDownSum);
       
        if (nMid < nBeg)
                return _inner_judge(2*nIndex+2, nBeg, nEnd, n, nPassDownSum);

        return _inner_judge(2*nIndex+1, nBeg, nMid, n, nPassDownSum) && _inner_judge(2*nIndex+2, nMid+1, nEnd, n, nPassDownSum);
}

void CTicketSystem::_inner_add(int nIndex, int nBeg, int nEnd, int n)
{
        if (m_pTree[nIndex].nFrom == nBeg && m_pTree[nIndex].nTo == nEnd)
        {
                m_pTree[nIndex].nMaxTickets += n;
                m_pTree[nIndex].nPassDown += n;
                return;
        }

        int nMid = (m_pTree[nIndex].nFrom + m_pTree[nIndex].nTo)/2;
        if (nMid >= nEnd)
        {
                _inner_add(2*nIndex+1, nBeg, nEnd, n);
                m_pTree[nIndex].nMaxTickets =
                        m_pTree[nIndex].nMaxTickets > m_pTree[2*nIndex+1].nMaxTickets ? m_pTree[nIndex].nMaxTickets : m_pTree[2*nIndex+1].nMaxTickets;
                return;
        }

        if (nMid < nBeg)
        {
                _inner_add(2*nIndex+2, nBeg, nEnd, n);
                m_pTree[nIndex].nMaxTickets =
                        m_pTree[nIndex].nMaxTickets > m_pTree[2*nIndex+2].nMaxTickets ? m_pTree[nIndex].nMaxTickets : m_pTree[2*nIndex+2].nMaxTickets;
                return;
        }

        _inner_add(2*nIndex+1, nBeg, nMid, n);
        _inner_add(2*nIndex+2, nMid+1, nEnd, n);

        m_pTree[nIndex].nMaxTickets =
                m_pTree[nIndex].nMaxTickets > m_pTree[2*nIndex+1].nMaxTickets ? m_pTree[nIndex].nMaxTickets : m_pTree[2*nIndex+1].nMaxTickets;
        m_pTree[nIndex].nMaxTickets =
                m_pTree[nIndex].nMaxTickets > m_pTree[2*nIndex+2].nMaxTickets ? m_pTree[nIndex].nMaxTickets : m_pTree[2*nIndex+2].nMaxTickets;
}
回复

使用道具 举报

🔗
bignews 2011-6-11 14:17:55 | 只看该作者
全局:
膜拜线段树……这玩意总是学不会啊
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 注册账号
隐私提醒:
  • ☑ 禁止发布广告,拉群,贴个人联系方式:找人请去🔗同学同事飞友,拉群请去🔗拉群结伴,广告请去🔗跳蚤市场,和 🔗租房广告|找室友
  • ☑ 论坛内容在发帖 30 分钟内可以编辑,过后则不能删帖。为防止被骚扰甚至人肉,不要公开留微信等联系方式,如有需求请以论坛私信方式发送。
  • ☑ 干货版块可免费使用 🔗超级匿名:面经(美国面经、中国面经、数科面经、PM面经),抖包袱(美国、中国)和录取汇报、定位选校版
  • ☑ 查阅全站 🔗各种匿名方法

本版积分规则

>
快速回复 返回顶部 返回列表