Lemonade Trade

input/code.cpp: In function 'int main()':
input/code.cpp:30:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < offers.size(); ++i) {
                                  ^
input/code.cpp:69:35: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for (int i = 0; i < revised.size(); ++i) {
                                   ^

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <utility>
#include <math.h>
#include <iomanip>

const long double inf = 1e9;

int main() {
	int n;
	std::cin >> n;
	std::vector<std::pair<std::pair<std::string, std::string>, long double>> offers;
	std::vector<std::pair<std::pair<int, int>, long double>> revised;
	std::vector<std::string> strings;
	strings.push_back("blue");
	strings.push_back("pink");
	for (int i = 0; i < n; ++i) {
		std::string a, b;
		long double ratio;
		std::cin >> a >> b >> ratio;
		strings.push_back(a);
		strings.push_back(b);
		offers.push_back({{a, b}, ratio});
	}
	std::sort(strings.begin(), strings.end());
	strings.erase(std::unique(strings.begin(), strings.end()), strings.end());
	int m = strings.size();
	for (int i = 0; i < offers.size(); ++i) {
		int low = 0;
		int high = m - 1;
		while(low != high) {
			int mid = (low + high) / 2;
			if (strings[mid] < offers[i].first.first) {
				low = mid + 1;
			} else {
				high = mid;
			}
		}
		int first = low;
		low = 0;
		high = m - 1;
		while(low != high) {
			int mid = (low + high) / 2;
			if (strings[mid] < offers[i].first.second) {
				low = mid + 1;
			} else {
				high = mid;
			}
		}
		int second = low;
		revised.push_back({{first, second}, offers[i].second});
	}
	int pink = -1;
	int blue = -1;
	for (int i = 0; i < m; ++i) {
		if (strings[i] == "pink") {
			pink = i;
		} else if (strings[i] == "blue") {
			blue = i;
		}
	}
	std::vector<long double> amounts (m);
	for (int i = 0; i < m; ++i) {
		amounts[i] = -inf;
	}
	amounts[pink] = 0;
	for (int i = 0; i < revised.size(); ++i) {
		int o = revised[i].first.first;
		int w = revised[i].first.second;
		// std::cout << o << ' ' << w << ' ' << amounts[o] << ' ' << amounts[w] << ' ' << log(revised[i].second) << '\n';
		amounts[o] = std::max(amounts[o], amounts[w] + log(revised[i].second));
	}
	/*
	std::cout << pink << ' ' << blue << '\n';
	for (int i = 0; i < m; ++i) {
		std::cout << amounts[i] << ' ';
	}
	std::cout << '\n';
	*/
	if (amounts[blue] > log(10)) {
		std::cout << 10 << '\n';
	} else {
		std::cout << std::fixed << std::setprecision(20) << exp(amounts[blue]) << '\n';
	}
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

Test 6

Verdict: ACCEPTED

Test 7

Verdict: ACCEPTED

Test 8

Verdict: ACCEPTED

Test 9

Verdict: ACCEPTED

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED