Cod sursa(job #882177)

Utilizator maritimCristian Lambru maritim Data 18 februarie 2013 22:16:03
Problema Zota & Chidil Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.98 kb
#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;

FILE *f = fopen("zc.in","r");
FILE *g = fopen("zc.out","w");

typedef vector<pair<int,int> >::iterator it;

#define MaxN 1001000
#define MaxTraps (13*MaxN)
#define MH 666013
#define mp make_pair
#define mij ((li+ls)>>1)
#define ll long long

int N,M;
ll Sol;
vector<pair<int,int> > A,B,C,D;

inline int abs(int a)
{
    return a < 0 ? -a : a;
}

inline void addTraps(int a,int b)
{
    for(int i=-2;i<=2;i++)
        C.push_back(mp(a,b+i));

    for(int i=-1;i<=1;i++)
        C.push_back(mp(a-1,b+i)),
        C.push_back(mp(a+1,b+i));

    C.push_back(mp(a-2,b));
    C.push_back(mp(a+2,b));
}

void citire(void)
{
    int a,b;

    fscanf(f,"%d %d\n",&N,&M);
    for(int i=1;i<=N;i++)
    {
        fscanf(f,"%d %d\n",&b,&a);
        addTraps(a,b);
    }
}

void creareTraps(void)
{
    sort(C.begin(),C.end());

    A.push_back(C[0]);

    for(int i=1;i<C.size();i++)
        if(C[i] != C[i-1] && (C[i].first || C[i].second))
            A.push_back(C[i]),
            B.push_back(mp(C[i].second,C[i].first));

     sort(B.begin(),B.end());
}

inline int cautBinx(int li,int ls,int a,int k)
{
    if(li > ls)
        return li;

    int x = (k == 0) ? B[mij].first : A[mij].first;

    if(x >= a)
        return cautBinx(li,mij-1,a,k);
    return cautBinx(mij+1,ls,a,k);
}

inline int cautBiny(int li,int ls,int a,int k)
{
    if(li > ls)
        return li;

    int x = (k == 0) ? B[mij].second : A[mij].second;

    if(x >= a)
        return cautBiny(li,mij-1,a,k);
    return cautBiny(mij+1,ls,a,k);
}

void Rezolvare(void)
{
    char op;
    int x,y,a,b,x1,x2,y1,y2,dx,dy;

    x = y = 0;

    creareTraps();

    for(int i=1;i<=M;i++)
    {
        fscanf(f,"%c %d\n",&op,&b);
        a = (op == 'E' || op == 'V');

        switch(op)
        {
            case 'N' : dx = b, dy = 0;
                break;
            case 'E' : dy = b, dx = 0;
                break;
            case 'S' : dx = -b, dy = 0;
                break;
            default : dy = -b, dx = 0;
        }

        if(a)
        {
            x1 = cautBinx(0,A.size()-1,x,a);
            x2 = cautBinx(0,A.size()-1,x+1,a);

            y1 = cautBiny(x1,x2-1,y,a);
            y2 = cautBiny(x1,x2-1,y+dy,a);

            //printf("%d ... %d %d /// y=%d y+dy=%d... %d %d -> ",x,x1,x2,y,y+dy,y1,y2);

        }
        else
        {
            x1 = cautBinx(0,A.size()-1,y,a);
            x2 = cautBinx(0,A.size()-1,y+1,a);

            y1 = cautBiny(x1,x2-1,x,a);
            y2 = cautBiny(x1,x2-1,x+dx,a);

            //printf("%d ... %d %d /// x=%d x+dx=%d %d %d -> ",y,x1,x2,x,x+dx,y1,y2);
        }

        //printf("%d %d ..... %d %d\n",x,y,x+dx,y+dy);

        x += dx;
        y += dy;

        Sol += abs(y2-y1);
    }
    
    //for(int i=0;i<A.size();i++)
    //    printf("%d %d\n",A[i].first,A[i].second);

    //for(int i=0;i<A.size();i++)
    //    printf("%d %d\n",B[i].first,B[i].second);
}

int main()
{
    citire();
    Rezolvare();

    fprintf(g,"%lld\n",Sol);
}