Artifact Content
Not logged in

Artifact 4e3a76b4d8344d4dae7021476215ec841b507d93


#include <iostream>
#include <sstream>
#include <iomanip>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <algorithm>
#include <numeric>
#include <iterator>
#include <functional>
#include <complex>
#include <queue>
#include <stack>
#include <cmath>
#include <cassert>
#include <cstring>
using namespace std;
typedef long long LL;
typedef complex<double> CMP;

template<typename T>
class IdGen
{
	map<T, int> v2id_;
	vector<T>   id2v_;
public:
	int v2id(const T& v) {
		if( !v2id_.count(v) ) { v2id_[v] = size(); id2v_.push_back(v); }
		return v2id_[v];
	}
	const T& id2v(int i) const { return id2v_[i]; }
	int size() const { return id2v_.size(); }
};

template<typename Vert, typename Cost, typename Flow, int NV=256>
class MinCostFlow
{
	IdGen<Vert> idgen;

	vector<int> G[NV];
	Flow F[NV][NV];
	Cost C[NV][NV];

public:
	void addEdge( Vert s_, Vert t_, Cost c, Flow f )
	{
		int s = idgen.v2id(s_), t = idgen.v2id(t_);
		G[s].push_back(t);
		G[t].push_back(s);
		C[s][t] = c;
		C[t][s] = -c;
		F[s][t] = f;
		F[t][s] = 0;
	}

	pair<Cost, Flow> calc( Vert s_, Vert t_ )
	{
		const int N=idgen.size(), S=idgen.v2id(s_), T=idgen.v2id(t_);
		static const Cost COST_INF = 1e+300; // !!EDIT HERE!!
		static const Flow FLOW_INF = 0x7fffffff;

		Cost total_cost = 0;
		Flow total_flow = 0;
		vector<Cost> dist(N, 0); // Distance from S : initially unknown.
		for(;;)
		{
			// Dijkstra : find the "shortest path" from S to T wrt C[][].
			//   C[][] can be <0 so we must be careful. Instead of computing the shortest path directly,
			//   we compute the increase ("delta") from the shortest path in the previous iteration.
			//   Since shortest path cannot decrease, delta is always >=0 when traversing edges.
			//   Smallest delta implies smallest dist[T]+delta[T].
			vector<Cost> delta(N, COST_INF); delta[S] = 0;
			vector<int>  prev(N, -1);

			typedef pair< Cost, pair<int, int> > cedge;
			priority_queue< cedge, vector<cedge>, greater<cedge> > Q;
			Q.push( cedge(0, make_pair(S, S)) );
			while( !Q.empty() ) {
				const cedge e = Q.top(); Q.pop();
				const int u_prev = e.second.first;
				const int u = e.second.second;
				if( prev[u] >= 0 ) // visited
					continue;
				prev[u] = u_prev;

				for(int i=0; i<G[u].size(); ++i) {
					const int  v = G[u][i];
					const Cost v_delta = dist[u]+delta[u]+C[u][v] - dist[v];
					if( F[u][v]>0 && delta[v]>v_delta )
						Q.push( cedge(delta[v]=v_delta, make_pair(u,v)) );
				}
			}

			// If T is unreachable, finished.
			if( prev[T] < 0 )
				break;

			// Update the distance table.
			for(int u=0; u<N; ++u)
				if( delta[u] != COST_INF )
					dist[u] += delta[u];

			// How much water can flow on the min-cost path?
			Flow f = FLOW_INF;
			for(int u=T; u!=S; u=prev[u])
				f = min(f, F[prev[u]][u]);

			// Run the flow as much as possible
			total_flow += f;
			for(int u=T; u!=S; u=prev[u]) {
				total_cost    += f * C[prev[u]][u];
				F[prev[u]][u] -= f;
				F[u][prev[u]] += f;
			}
		}
		return make_pair(total_cost, total_flow);
	}
};

class FoxCardGame { public:
	double theMaxProportion(vector <double> pileA, vector <double> pileB, int k) 
	{
		double L=1, R=50;
		for(int i=0; i<50; ++i)
			(possible(pileA, pileB, k, (L+R)/2) ? L : R) = (L+R)/2;
		return L;
	}

	enum Tag {Src, Left, Right, Target, Goal};
	bool possible(vector <double> pileA, vector <double> pileB, int k, double R)
	{
		MinCostFlow<pair<Tag,int>, double, int> mcf;

		for(int a=0; a<pileA.size(); ++a)
			mcf.addEdge( make_pair(Src,0), make_pair(Left,a), 0.0, 1 );
		for(int a=0; a<pileA.size(); ++a)
		for(int b=0; b<pileB.size(); ++b)
		{
			double x_ = pileA[a]+pileB[b], y_ = pileA[a]*pileB[b];
			double x = max(x_,y_), y = min(x_,y_);
			mcf.addEdge( make_pair(Left,a), make_pair(Right,b), R*y-x+10000.0, 1 );
		}
		for(int b=0; b<pileB.size(); ++b)
			mcf.addEdge( make_pair(Right,b), make_pair(Target,0), 0.0, 1 );
		mcf.addEdge( make_pair(Target,0), make_pair(Goal,0), 0.0, k );

		return mcf.calc( make_pair(Src,0), make_pair(Goal,0) ).first <= 10000.0*k;
	}
};

// BEGIN CUT HERE
#include <ctime>
double start_time; string timer()
 { ostringstream os; os << " (" << int((clock()-start_time)/CLOCKS_PER_SEC*1000) << " msec)"; return os.str(); }
template<typename T> ostream& operator<<(ostream& os, const vector<T>& v)
 { os << "{ ";
   for(typename vector<T>::const_iterator it=v.begin(); it!=v.end(); ++it)
   os << '\"' << *it << '\"' << (it+1==v.end() ? "" : ", "); os << " }"; return os; }
void verify_case(const double& Expected, const double& Received) {
 bool ok = (abs(Expected - Received) < 1e-9);
 if(ok) cerr << "PASSED" << timer() << endl;  else { cerr << "FAILED" << timer() << endl;
 cerr << "\to: \"" << Expected << '\"' << endl << "\tx: \"" << Received << '\"' << endl; } }
#define CASE(N) {cerr << "Test Case #" << N << "..." << flush; start_time=clock();
#define END	 verify_case(_, FoxCardGame().theMaxProportion(pileA, pileB, k));}
int main(){

CASE(0)
	double pileA_[] = {1, 2, 3};
	  vector <double> pileA(pileA_, pileA_+sizeof(pileA_)/sizeof(*pileA_)); 
	double pileB_[] = {4, 5, 6};
	  vector <double> pileB(pileB_, pileB_+sizeof(pileB_)/sizeof(*pileB_)); 
	int k = 2; 
	double _ = 1.7692307692309948; 
END
CASE(1)
	double pileA_[] = {1.234, 5.678, 9.012, 3.456, 7.89};
	  vector <double> pileA(pileA_, pileA_+sizeof(pileA_)/sizeof(*pileA_)); 
	double pileB_[] = {2.345, 6.789, 9.876, 5.432, 1.012};
	  vector <double> pileB(pileB_, pileB_+sizeof(pileB_)/sizeof(*pileB_)); 
	int k = 3; 
	double _ = 4.159424420079586; 
END
CASE(2)
	double pileA_[] = {1, 1.1, 1.2, 1.3, 1.4, 1.5};
	  vector <double> pileA(pileA_, pileA_+sizeof(pileA_)/sizeof(*pileA_)); 
	double pileB_[] = {5, 10, 15, 20, 25, 30};
	  vector <double> pileB(pileB_, pileB_+sizeof(pileB_)/sizeof(*pileB_)); 
	int k = 2; 
	double _ = 1.3972602739726827; 
END
CASE(3)
	double pileA_[] = {85.302, 92.798, 76.813, 37.994, 36.737, 98.659};
	  vector <double> pileA(pileA_, pileA_+sizeof(pileA_)/sizeof(*pileA_)); 
	double pileB_[] = {13.352, 7.3094, 54.761, 8.2706, 63.223, 37.486};
	  vector <double> pileB(pileB_, pileB_+sizeof(pileB_)/sizeof(*pileB_)); 
	int k = 3; 
	double _ = 33.58603889836175; 
END
CASE(4)
	double pileA_[] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0};
	  vector <double> pileA(pileA_, pileA_+sizeof(pileA_)/sizeof(*pileA_)); 
	double pileB_[] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0};
	  vector <double> pileB(pileB_, pileB_+sizeof(pileB_)/sizeof(*pileB_)); 
	int k = 50; 
	double _ = 16.846938775510203; 
END
CASE(5)
	double pileA_[] = {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0};
	  vector <double> pileA(pileA_, pileA_+sizeof(pileA_)/sizeof(*pileA_)); 
	double pileB_[] = {51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0};
	  vector <double> pileB(pileB_, pileB_+sizeof(pileB_)/sizeof(*pileB_)); 
	int k = 50; 
	double _ = 21.128144186967717; 
END
}
// END CUT HERE