Tuesday, August 26, 2014

Getting the Center of a Tree


Credit goes to: Ahmed Ossama, YellowNextYear (codeforces)
 vector<int> getRoots(int n, vector<vector<int> > &children) {
  children.clear();
  children.resize(n);
  vector<int> cnt(n, 0);
  queue<int> Q;
  vector<bool> v(n, 0);
  int x, y, count;
  for (int i = 0; i < n - 1; i++) {
   cin >> x >> y;
   x--;
   y--;
   children[x].pb(y);
   children[y].pb(x);
   cnt[x]++;
   cnt[y]++;
  }
  for (int i = 0; i < n; i++)
   if (cnt[i] == 1)
    Q.push(i);
  count = 0;
  while (!Q.empty()) {
   x = Q.front();
   Q.pop();
   count++;
   v[x] = 1;
   for (int i = 0; i < children[x].size(); i++) {
    y = children[x][i];
    if (cnt[y] == 0) {
     continue;
    }
    cnt[x]--;
    cnt[y]--;
    if (cnt[y] == 1) {
     Q.push(y);
    }
   }
   if (count == n - 2)
    break;
  }
  while (!Q.empty())
   Q.pop();
  vector<int> ret;
  for (int i = 0; i < n; i++) {
   if (v[i] == 0) {
    ret.pb(i);
   }
  }
  return ret;
 }
my method of getting the centers:
bool getRoots(int cur, int parent, int target, int depth, int&length, vector<int>& roots, vector<vector<int> >&t){
 if(target == cur){
  length = depth;
  return true;
 }
 rep(i,sz(t[cur])){
  if(t[cur][i] == parent) continue;
  if(getRoots(t[cur][i],cur,target,depth+1,length,roots,t)){
   if(length/2 == depth || (length+1)/2 == depth)
    roots.pb(cur);
   return true;
  }
 }
 return false;
}
int getFarthest(int cur, vector<vector<int> >&t, int&far,int parent = -1){
 if(parent != -1 && sz(t[cur]) == 1) {
  far = cur;
  return 0;
 }
 int mxDepth = -1, idx = -1;
 rep(i,sz(t[cur])){
  if(parent == t[cur][i]) continue;
  int d, f;
  d = getFarthest(t[cur][i],t,f,cur);
  if(d > mxDepth){
   mxDepth = d;
   idx = f;
  }
 }
 far = idx;
 return mxDepth;
}
vector<int> getRoots(vector<vector<int> >&t){
 int v1, v2;
 getFarthest(0, t, v1);
 getFarthest(v1,t, v2);
 int length = 0;
 vector<int> ret;
 getRoots(v1,-1,v2,0,length,ret,t);
 return ret;
}

Monday, August 25, 2014

Checking if 2 trees are similar (Trees Isomorphism)


Check this PDF, but be careful because it has some minor mistakes: link Source: http://www14.informatik.tu-muenchen.de/konferenzen/Jass08/courses/1/smal/Smal_Paper.pdf

Sunday, August 24, 2014

Find Distance To Segment

Source: http://www.cprogramto.com/c-program-to-find-shortest-distance-between-point-and-line-segment/
double FindDistanceToSegment(double x1, double y1, double x2, double y2, double pointX, double pointY)
{
    double diffX = x2 - x1;
    double diffY = y2 - y1;
    if ((diffX == 0) && (diffY == 0))
    {
        diffX = pointX - x1;
        diffY = pointY - y1;
        return sqrt(diffX * diffX + diffY * diffY);
    }

    double t = ((pointX - x1) * diffX + (pointY - y1) * diffY) / (diffX * diffX + diffY * diffY);

    if (t < 0)
    {
        //point is nearest to the first point i.e x1 and y1
        diffX = pointX - x1;
        diffY = pointY - y1;
    }
    else if (t > 1)
    {
        //point is nearest to the end point i.e x2 and y2
        diffX = pointX - x2;
        diffY = pointY - y2;
    }
    else
    {
        //if perpendicular line intersect the line segment.
        diffX = pointX - (x1 + t * diffX);
        diffY = pointY - (y1 + t * diffY);
    }

    //returning shortest distance
    return sqrt(diffX * diffX + diffY * diffY);
}

Thursday, August 21, 2014

C++ BigInt for Sport Programmers in ACM contests


source: http://pastebin.com/aQ8NJ197 Special Thanks to: SkorKNURE (on CodeForces)
#include <vector>
#include <cstdlib>
#include <iostream>
#include <iomanip>
#include <string>
#include <ctime>
using namespace std;
 
// base and base_digits must be consistent
const int base = 1000000000;
const int base_digits = 9;
 
struct bigint {
    vector<int> a;
    int sign;
 
    bigint() :
        sign(1) {
    }
 
    bigint(long long v) {
        *this = v;
    }
 
    bigint(const string &s) {
        read(s);
    }
 
    void operator=(const bigint &v) {
        sign = v.sign;
        a = v.a;
    }
 
    void operator=(long long v) {
        sign = 1;
        if (v < 0)
            sign = -1, v = -v;
        for (; v > 0; v = v / base)
            a.push_back(v % base);
    }
 
    bigint operator+(const bigint &v) const {
        if (sign == v.sign) {
            bigint res = v;
 
            for (int i = 0, carry = 0; i < (int) max(a.size(), v.a.size()) || carry; ++i) {
                if (i == (int) res.a.size())
                    res.a.push_back(0);
                res.a[i] += carry + (i < (int) a.size() ? a[i] : 0);
                carry = res.a[i] >= base;
                if (carry)
                    res.a[i] -= base;
            }
            return res;
        }
        return *this - (-v);
    }
 
    bigint operator-(const bigint &v) const {
        if (sign == v.sign) {
            if (abs() >= v.abs()) {
                bigint res = *this;
                for (int i = 0, carry = 0; i < (int) v.a.size() || carry; ++i) {
                    res.a[i] -= carry + (i < (int) v.a.size() ? v.a[i] : 0);
                    carry = res.a[i] < 0;
                    if (carry)
                        res.a[i] += base;
                }
                res.trim();
                return res;
            }
            return -(v - *this);
        }
        return *this + (-v);
    }
 
    void operator*=(int v) {
        if (v < 0)
            sign = -sign, v = -v;
        for (int i = 0, carry = 0; i < (int) a.size() || carry; ++i) {
            if (i == (int) a.size())
                a.push_back(0);
            long long cur = a[i] * (long long) v + carry;
            carry = (int) (cur / base);
            a[i] = (int) (cur % base);
            //asm("divl %%ecx" : "=a"(carry), "=d"(a[i]) : "A"(cur), "c"(base));
        }
        trim();
    }
 
    bigint operator*(int v) const {
        bigint res = *this;
        res *= v;
        return res;
    }
 
    friend pair<bigint, bigint> divmod(const bigint &a1, const bigint &b1) {
        int norm = base / (b1.a.back() + 1);
        bigint a = a1.abs() * norm;
        bigint b = b1.abs() * norm;
        bigint q, r;
        q.a.resize(a.a.size());
 
        for (int i = a.a.size() - 1; i >= 0; i--) {
            r *= base;
            r += a.a[i];
            int s1 = r.a.size() <= b.a.size() ? 0 : r.a[b.a.size()];
            int s2 = r.a.size() <= b.a.size() - 1 ? 0 : r.a[b.a.size() - 1];
            int d = ((long long) base * s1 + s2) / b.a.back();
            r -= b * d;
            while (r < 0)
                r += b, --d;
            q.a[i] = d;
        }
 
        q.sign = a1.sign * b1.sign;
        r.sign = a1.sign;
        q.trim();
        r.trim();
        return make_pair(q, r / norm);
    }
 
    bigint operator/(const bigint &v) const {
        return divmod(*this, v).first;
    }
 
    bigint operator%(const bigint &v) const {
        return divmod(*this, v).second;
    }
 
    void operator/=(int v) {
        if (v < 0)
            sign = -sign, v = -v;
        for (int i = (int) a.size() - 1, rem = 0; i >= 0; --i) {
            long long cur = a[i] + rem * (long long) base;
            a[i] = (int) (cur / v);
            rem = (int) (cur % v);
        }
        trim();
    }
 
    bigint operator/(int v) const {
        bigint res = *this;
        res /= v;
        return res;
    }
 
    int operator%(int v) const {
        if (v < 0)
            v = -v;
        int m = 0;
        for (int i = a.size() - 1; i >= 0; --i)
            m = (a[i] + m * (long long) base) % v;
        return m * sign;
    }
 
    void operator+=(const bigint &v) {
        *this = *this + v;
    }
    void operator-=(const bigint &v) {
        *this = *this - v;
    }
    void operator*=(const bigint &v) {
        *this = *this * v;
    }
    void operator/=(const bigint &v) {
        *this = *this / v;
    }
 
    bool operator<(const bigint &v) const {
        if (sign != v.sign)
            return sign < v.sign;
        if (a.size() != v.a.size())
            return a.size() * sign < v.a.size() * v.sign;
        for (int i = a.size() - 1; i >= 0; i--)
            if (a[i] != v.a[i])
                return a[i] * sign < v.a[i] * sign;
        return false;
    }
 
    bool operator>(const bigint &v) const {
        return v < *this;
    }
    bool operator<=(const bigint &v) const {
        return !(v < *this);
    }
    bool operator>=(const bigint &v) const {
        return !(*this < v);
    }
    bool operator==(const bigint &v) const {
        return !(*this < v) && !(v < *this);
    }
    bool operator!=(const bigint &v) const {
        return *this < v || v < *this;
    }
 
    void trim() {
        while (!a.empty() && !a.back())
            a.pop_back();
        if (a.empty())
            sign = 1;
    }
 
    bool isZero() const {
        return a.empty() || (a.size() == 1 && !a[0]);
    }
 
    bigint operator-() const {
        bigint res = *this;
        res.sign = -sign;
        return res;
    }
 
    bigint abs() const {
        bigint res = *this;
        res.sign *= res.sign;
        return res;
    }
 
    long long longValue() const {
        long long res = 0;
        for (int i = a.size() - 1; i >= 0; i--)
            res = res * base + a[i];
        return res * sign;
    }
 
    friend bigint gcd(const bigint &a, const bigint &b) {
        return b.isZero() ? a : gcd(b, a % b);
    }
    friend bigint lcm(const bigint &a, const bigint &b) {
        return a / gcd(a, b) * b;
    }
 
    void read(const string &s) {
        sign = 1;
        a.clear();
        int pos = 0;
        while (pos < (int) s.size() && (s[pos] == '-' || s[pos] == '+')) {
            if (s[pos] == '-')
                sign = -sign;
            ++pos;
        }
        for (int i = s.size() - 1; i >= pos; i -= base_digits) {
            int x = 0;
            for (int j = max(pos, i - base_digits + 1); j <= i; j++)
                x = x * 10 + s[j] - '0';
            a.push_back(x);
        }
        trim();
    }
 
    friend istream& operator>>(istream &stream, bigint &v) {
        string s;
        stream >> s;
        v.read(s);
        return stream;
    }
 
    friend ostream& operator<<(ostream &stream, const bigint &v) {
        if (v.sign == -1)
            stream << '-';
        stream << (v.a.empty() ? 0 : v.a.back());
        for (int i = (int) v.a.size() - 2; i >= 0; --i)
            stream << setw(base_digits) << setfill('0') << v.a[i];
        return stream;
    }
 
    static vector<int> convert_base(const vector<int> &a, int old_digits, int new_digits) {
        vector<long long> p(max(old_digits, new_digits) + 1);
        p[0] = 1;
        for (int i = 1; i < (int) p.size(); i++)
            p[i] = p[i - 1] * 10;
        vector<int> res;
        long long cur = 0;
        int cur_digits = 0;
        for (int i = 0; i < (int) a.size(); i++) {
            cur += a[i] * p[cur_digits];
            cur_digits += old_digits;
            while (cur_digits >= new_digits) {
                res.push_back(int(cur % p[new_digits]));
                cur /= p[new_digits];
                cur_digits -= new_digits;
            }
        }
        res.push_back((int) cur);
        while (!res.empty() && !res.back())
            res.pop_back();
        return res;
    }
 
    typedef vector<long long> vll;
 
    static vll karatsubaMultiply(const vll &a, const vll &b) {
        int n = a.size();
        vll res(n + n);
        if (n <= 32) {
            for (int i = 0; i < n; i++)
                for (int j = 0; j < n; j++)
                    res[i + j] += a[i] * b[j];
            return res;
        }
 
        int k = n >> 1;
        vll a1(a.begin(), a.begin() + k);
        vll a2(a.begin() + k, a.end());
        vll b1(b.begin(), b.begin() + k);
        vll b2(b.begin() + k, b.end());
 
        vll a1b1 = karatsubaMultiply(a1, b1);
        vll a2b2 = karatsubaMultiply(a2, b2);
 
        for (int i = 0; i < k; i++)
            a2[i] += a1[i];
        for (int i = 0; i < k; i++)
            b2[i] += b1[i];
 
        vll r = karatsubaMultiply(a2, b2);
        for (int i = 0; i < (int) a1b1.size(); i++)
            r[i] -= a1b1[i];
        for (int i = 0; i < (int) a2b2.size(); i++)
            r[i] -= a2b2[i];
 
        for (int i = 0; i < (int) r.size(); i++)
            res[i + k] += r[i];
        for (int i = 0; i < (int) a1b1.size(); i++)
            res[i] += a1b1[i];
        for (int i = 0; i < (int) a2b2.size(); i++)
            res[i + n] += a2b2[i];
        return res;
    }
 
    bigint operator*(const bigint &v) const {
        vector<int> a6 = convert_base(this->a, base_digits, 6);
        vector<int> b6 = convert_base(v.a, base_digits, 6);
        vll a(a6.begin(), a6.end());
        vll b(b6.begin(), b6.end());
        while (a.size() < b.size())
            a.push_back(0);
        while (b.size() < a.size())
            b.push_back(0);
        while (a.size() & (a.size() - 1))
            a.push_back(0), b.push_back(0);
        vll c = karatsubaMultiply(a, b);
        bigint res;
        res.sign = sign * v.sign;
        for (int i = 0, carry = 0; i < (int) c.size(); i++) {
            long long cur = c[i] + carry;
            res.a.push_back((int) (cur % 1000000));
            carry = (int) (cur / 1000000);
        }
        res.a = convert_base(res.a, 6, base_digits);
        res.trim();
        return res;
    }
};
 
int main() {
    bigint a("99999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999");
    bigint b("19999999999999999999999999999999999999999999999999999999999999999999999999999999999999998");
    cout << a * b << endl;
    cout << a / b << endl;
 
    string sa, sb;
    for (int i = 0; i < 100000; i++)
        sa += i % 10 + '0';
    for (int i = 0; i < 20000; i++)
        sb += i % 10 + '0';
    a = bigint(sa);
    b = bigint(sb);
 
    clock_t start = clock();
    bigint c = a / b;
    fprintf(stderr, "time=%.3lfsec\n", 0.001 * (clock() - start));
}

Friday, August 15, 2014

Gaussian Elimination


#include <vector>
#include <algorithm>
#include <iostream>
#include <cmath>
using namespace std;
const double eps = 1e-9;
enum sol {
 NOSOL, UNIQUE, INF
};
sol GaussianElimination(vector<vector<double> >& lhsMat, vector<double>& rhs,
  vector<double>&ans) {
 // the final answer will be in the ans
 int eqnCnt = (int) lhsMat.size(), varCnt = (int) lhsMat[0].size();
 sol ret = UNIQUE;
 for (int curCol = 0, curRow = 0; curCol < min(eqnCnt, varCnt);
   curCol++, curRow++) {
  for (int j = curRow; j < eqnCnt; j++) { // get a row where mat[i][i] is not zero
   if (fabs(lhsMat[j][curCol]) > eps) {
    swap(lhsMat[curRow], lhsMat[j]); // swapping vectors O(1)
    swap(rhs[curRow], rhs[j]);
    break;
   }
  }
  if (fabs(lhsMat[curRow][curCol]) < eps) { // if the selected cell is still 0, stay in the same row.
   curRow--;
   continue;
  }

  // make all cells below curCell 0
  for (int j = curRow + 1; j < eqnCnt; j++) {
   if (fabs(lhsMat[j][curCol]) < eps)
    continue;
   double mulByThis = -lhsMat[j][curCol] / lhsMat[curRow][curCol];
   for (int k = curCol; k < varCnt; k++) {
    lhsMat[j][k] += mulByThis * lhsMat[curRow][k];
   }
   rhs[j] += mulByThis * rhs[curRow];
  }
 }
 int calculatedValues = 0;
 ans = vector<double>(varCnt, 0.0);
 // go from bottom to top and get the answers
 for (int i = eqnCnt - 1; i >= 0; --i) {
  int k = 0;
  for (k = 0; k < varCnt; k++) {
   if (fabs(lhsMat[i][k]) > eps) {
    break;
   }
  }
  if (k == varCnt) {
   if (fabs(rhs[i]) > eps)
    return NOSOL;
   continue;
  }
  double val = rhs[i];
  for (int j = k + 1; j < varCnt; j++) {
   val -= ans[j] * lhsMat[i][j];
  }
  val /= lhsMat[i][k];

  ans[k] = val;
  calculatedValues++;
 }
 if (calculatedValues != varCnt)
  ret = INF;
 return ret;
}
int main(){
 int n, m;
 cin >> n >> m;
 vector<vector<double> > lhsMat(n, vector<double>(m));
 vector<double> rhs(n), ans;
 for(int i = 0; i < n; i++){
  for(int k = 0; k < m; k++){
   cin >> lhsMat[i][k];
  }
  cin >> rhs[i];
 }
 GaussianElimination(lhsMat,rhs,ans);
 for(int i = 0; i < (int) ans.size(); i++){
  cout << ans[i] << endl;
 }
 return 0;
}

Friday, August 8, 2014

Solution for Live Archive 10819 - Trouble of 13-Dots


#include <bits/stdc++.h>
using namespace std;
//const int Maxsize=102;
int mem[101][10201];
int P[101];
int F[101];
bool Money[101];
int M,N;
bool flag;
int dp(int in,int Spent)
{
// if(Spent > M && M<1800)  return 0;
// if(Spent > m+200) return 0;
 if(M > 1800){
  if(Spent > M+200) return -1000000;
  if(in == N){
   if( Spent > M && Spent <= 2000) return -1000000;
   else return 0;
  }
 } else {
  if(Spent > M) return -1000000;
  if(in ==N) return 0;
 }

 int &ret=mem[in][Spent];
 if(ret!= -1)
  return ret;
 ret=dp(in+1,Spent);
 ret=max(dp(in+1,(Spent+P[in]))+F[in],ret);
 return ret;
}
int main()
{
// freopen("in.txt","rt",stdin);
 while(cin >> M >> N)
 {
  flag=true;
  memset(mem,-1,sizeof mem);
  for(int j=0 ; j<N ; j++)
  {
   cin>>P[j]>>F[j];
  }
  cout<<dp(0,0);
  cout<<'\n';
 }
 return 0;
}

Wednesday, August 6, 2014

Solution for Timus 1992. CVS


/*
 * 1992.cpp
 *
 *  Created on: Aug 5, 2014
 *      Author: ghooo
 */

#include <cstring>
#include <map>
#include <deque>
#include <queue>
#include <stack>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <algorithm>
#include <vector>
#include <set>
#include <complex>
#include <list>
#include <climits>
#include <cctype>
#include <bitset>
//#include <windows.h>

using namespace std;

#define pb push_back
#define all(v) v.begin(),v.end()
#define rall(v) v.rbegin(),v.rend()
#define sz(v) ((ll)v.size())
#define rep(i,m) for(int i=0;i<(int)(m);++i)
#define repe(i,n,m) for(int i=n;i<=(int)(m);++i)
#define rep2(i,n,m) for(int i=n;i<(int)(m);i++)
#define For(it,c) for(__typeof(c.begin()) it=c.begin();it!=c.end();++it)
#define reset(a,b) memset(a,b,sizeof(a))
#define mp make_pair
#define dot(a,b) ((conj(a)*(b)).X)
#define X real()
#define Y imag()
#define length(V) (hypot((V).X,(V).Y))
#define vect(a,b) ((b)-(a))
#define cross(a,b) ((conj(a)*(b)).imag())
#define normalize(v) ((v)/length(v))
#define rotate(p,about,theta) ((p-about)*exp(point(0,theta))+about)
#define pointEqu(a,b) (comp(a.X,b.X)==0 && comp(a.Y,b.Y)==0)
#define clrq(x) while(!x.empty()) x.pop();
#define clrvv(v) rep(i,sz(v))v[i].clear();
#define debug(x) cerr << #x << ": " << x << endl;
#define debugv(v) cerr << #v << ": ";For(it,v)cerr <<(*it)<<", "; cerr<<endl;

typedef stringstream ss;
typedef long long ll;
typedef pair<int, int> pii;
typedef vector<pii> vpii;
typedef vector<string> vs;
typedef vector<int> vi;
typedef vector<double> vd;
typedef vector<vector<int> > vii;
typedef long double ld;
typedef complex<double> point;
typedef pair<point, point> segment;
typedef pair<double, point> circle;
typedef vector<point> polygon;
typedef unsigned long long ull;
typedef long long ll;

//==============================================================
// handling triples
typedef pair<ll,pair<ll,ll> > triple;
#define tfirst first
#define tsecond second.first
#define tthird second.second
#define mt(x,y,z) mp(x,mp(y,z))
//---------------------------------------------------------------

const int oo = (int) 1e9;
const double PI = 2 * acos(0.0);
const double eps = 1e-9;
vector<int> cloneProgram, cloneCancel;
vector<int> programHistory, programHistoryPrev;
vector<int> cancelHistory, cancelHistoryPrev;
void learn(int cl, int prog){
 programHistory.pb(prog);
 programHistoryPrev.pb(cloneProgram[cl]);
 cloneProgram[cl] = sz(programHistory)-1;
 cloneCancel[cl] = -1;
}
void rollback(int cl){
 cancelHistory.pb(cloneProgram[cl]);
 cancelHistoryPrev.pb(cloneCancel[cl]);
 cloneCancel[cl] = sz(cancelHistory)-1;

 cloneProgram[cl] = programHistoryPrev[cloneProgram[cl]];
}
void relearn(int cl){
 int lastProgram = cancelHistory[cloneCancel[cl]];

 cloneProgram[cl] = lastProgram;

 cloneCancel[cl] = cancelHistoryPrev[cloneCancel[cl]];

}
void clone(int cl){
 cloneProgram.pb(cloneProgram[cl]);
 cloneCancel.pb(cloneCancel[cl]);
}
void check(int cl){
 if(cloneProgram[cl] == -1) cout << "basic" << endl;
 else {
  cout << programHistory[cloneProgram[cl]] << endl;
 }
}
void init(){
 cloneProgram.clear();
 cloneProgram.pb(-1);
 cloneProgram.pb(-1);
 cloneCancel.clear();
 cloneCancel.pb(-1);
 cloneCancel.pb(-1);
 programHistory.clear();
 programHistoryPrev.clear();
 cancelHistory.clear();
 cancelHistoryPrev.clear();
}
int main(){
 ios_base::sync_with_stdio(false);
 cin.tie(NULL);
 cout.tie(NULL);
#ifndef ONLINE_JUDGE
 freopen("in.txt", "r", stdin);
 //freopen("out.txt", "w", stdout);
#endif
 int n, m;
 cin >> n >> m;
 init();
 string s;
 int c, p;
 rep(i,n){
  cin >> s;
  if(s == "learn"){
   cin >> c >> p;
   learn(c,p);
  } else if(s == "rollback"){
   cin >> c;
   rollback(c);
  } else if(s == "relearn"){
   cin >> c;
   relearn(c);
  } else if(s == "clone"){
   cin >> c;
   clone(c);
  } else if(s == "check"){
   cin >> c;
   check(c);
  }
//  debugv(cloneProgram);
//  debugv(cloneCancel);
//  debugv(programHistory);
//  debugv(pPrev);
//  debugv(cancelHistory);
//  debugv(cPrev);
 }
 return 0;
}