SRM635 div1 easy SimilarRatingGraph

解法

i,jから始まる部分列について同型である直線の長さを考える。各区間について傾きが等しく、比率が等しいならば同型であると判定できる。
i+k→i+k+1、j+k→j+k+1の線分について傾きが等しく、day[i+k+1]-day[i+k] と day[j+k+1] - day[j+k] の比率が等しければ部分列[i,i+k]と[j,j+k]は同型である。同型であるところまでkを0～nまで全て試す。
i,jについて全て試す部分でO(N^2)、どこまで同型なのかkを試す部分がO(N)で全体でO(N^3)で求められる。

double = int / int というポカで落として反省。0除算とか色々罠が多そう。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<ll> VL;
typedef vector<VL> VVL;
typedef pair<int, int> PII;

#define FOR(i, a, n) for (ll i = (ll)a; i < (ll)n; ++i)
#define REP(i, n) FOR(i, 0, n)
#define ALL(x) x.begin(), x.end()
#define IN(a, b, x) (a<=x&&x<b)
#define MP make_pair
#define PB push_back
const int INF = (1LL<<30);
const ll LLINF = (1LL<<60);
const double PI = 3.14159265359;
const double EPS = 1e-12;
const int MOD = 1000000007;
//#define int ll

template <typename T> T &chmin(T &a, const T &b) { return a = min(a, b); }
template <typename T> T &chmax(T &a, const T &b) { return a = max(a, b); }

int dx[] = {0, 1, 0, -1}, dy[] = {1, 0, -1, 0};

class SimilarRatingGraph {
   public:
   double maxLength(vector <int> date, vector <int> rating)
  {
    int n = date.size();
    double ret = 0.0;
    REP(i, n) FOR(j, i+1, n) {
      double ratio, len = 0;
      REP(k, n) {
        if(j+k+1 >= n) break;
        // i+k→i+k+1 と j+k→j+k+1 の比が等しいか？
        if((rating[i+k+1]-rating[i+k])*(date[j+k+1]-date[j+k]) == (rating[j+k+1]-rating[j+k])*(date[i+k+1]-date[i+k])) {
          if(k == 0) {
            ratio = (double)(date[j+k+1]-date[j+k])/(date[i+k+1]-date[i+k]);
          } else {
            if(ratio != (double)(date[j+k+1]-date[j+k])/(date[i+k+1]-date[i+k])) {
              break;
            }
          }
          if(ratio > 1) {
            len += sqrt(pow(rating[j+k+1]-rating[j+k], 2) + pow(date[j+k+1]-date[j+k], 2));
          } else {
            len += sqrt(pow(rating[i+k+1]-rating[i+k], 2) + pow(date[i+k+1]-date[i+k], 2));
          }
          chmax(ret, len);
        } else {
          break;
        }
      }
    }

    return ret;
  }
};