Cod sursa(job #2049305)

Utilizator b0gd4nBogdan b0gd4n Data 27 octombrie 2017 00:33:30
Problema Heavy metal Scor 60
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.67 kb
#include <iostream>
#include <stdio.h>
#include <vector>
#include <algorithm>

using namespace std;

FILE *f, *g;

struct concert
{
  int A, B;
};

int N, A, B;
int Tmax; //Timpul maxim de terminare a unui concert.

//Memorez intervalele concertelor.
vector<concert> C;

//Best[i]=timpul total maxim in care vor canta formatii daca fiecare concert se va termina inainte de ora i.
//vector<int> Best;
int Best[1<<17];

concert Make (int A, int B)
{
  concert r = {A, B};
  return r;
}

bool How (concert c1, concert c2)
{
  return c1.B < c2.B;
}

int main()
{
  f = fopen ("heavymetal.in", "r");
  g = fopen ("heavymetal.out", "w");

  fscanf (f, "%d", &N);
  for (int i = 1; i <= N; i++)
  {
    fscanf (f, "%d %d", &A, &B);
    C.push_back (Make (A, B) );
    Tmax = max (Tmax, B);
  }

  //Sortez crescator dupa campul B.
  sort (C.begin(), C.end(), How);

  //for (int i = 0; i < N; i++) cout << C[i].A << " " << C[i].B << endl;

  //Best.push_back (0);
  /*for (int i = 1; i <= Tmax; i++)
  {
    Best.push_back(Best[i-1]);
    for (int j = 0; j < N; j++)
    {
      if (C[j].B == i) Best[i] = max(Best[i], Best[C[j].A] + (C[j].B - C[j].A));
      else if (C[j].B > i) break;
    }
  }
  */
  Best[C[0].A] = 0;
  for (int k = 0; k < N; k++)
  {
    int i = C[k].B;
    //cout << i << endl;
    Best[i] = (k > 0) ? Best[C[k - 1].B] : 0;

    Best[i] = max (Best[i], Best[C[k].A] + (C[k].B - C[k].A) );
    if (i < C[k + 1].A && k + 1 < N) Best[C[k + 1].A] = Best[i];
  }
  //for (int i = 0; i <= Tmax; i++) cout << i << "=" << Best[i] << endl;

  fprintf (g, "%d", Best[Tmax]);

  fclose (f);
  fclose (g);

  return 0;
}