【NOIP2011提高组T6】观光公交-贪心

(本人本题完成于2016-7-21)

题目:观光公交题目

以下是本人代码:

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m,k,d[1010]={0},go[1010]={0},arrive[1010]={0},p[1010]={0}; //go[i],arrive[i]:公交在第i站的出发和到达时间
int s[10010]={0},t[10010]={0},a[10010]={0},ans=0; //a[i],s[i],t[i]:第i个乘客出现时间,出发站编号和终点站编号

void input()
{
  scanf("%d %d %d",&n,&m,&k);
  for(int i=1;i<=n-1;i++)
    scanf("%d",&d[i]);
  for(int i=1;i<=m;i++)
  {
    scanf("%d %d %d",&a[i],&s[i],&t[i]);
    if (a[i]>go[s[i]]) go[s[i]]=a[i];
    p[t[i]-1]++; //p[i]:在第i站使用加速器的收益
  }
}

void work()
{
  for(int i=1;i<=n;i++) arrive[i]=max(arrive[i-1],go[i-1])+d[i-1]; //计算未使用加速器时的到达时间
  for(int i=1;i<=k;i++)
  {
    int maxi=0,c=0,maxc=0;
    for(int j=n-1;j>=1;j--) //从后往前推,求出要在哪站使用加速器
    {
      if (arrive[j+1]<=go[j+1]) c=0; //如果到站时间比出发时间还早,则没有收益
      c+=p[j]; //增加当前站收益
      if (d[j]>0&&c>maxc) {maxi=j;maxc=c;}
    }
if (maxi==0) break;
    d[maxi]--;
    for(int j=maxi+1;j<=n;j++) arrive[j]=max(arrive[j-1],go[j-1])+d[j-1]; //更新使用加速器的站之后的到达时间
  }
  for(int i=1;i<=n;i++) arrive[i]=max(arrive[i-1],go[i-1])+d[i-1]; //最后再更新一遍,防止有未更新的站
  for(int i=1;i<=m;i++) ans+=arrive[t[i]]-a[i]; //累加答案
}

void output()
{
  printf("%d",ans);
}

int main()
{
  input();
  work();
  output();
  return 0;
}

posted @ 2016-07-21 22:19  Maxwei_wzj  阅读(118)  评论(0编辑  收藏  举报