小W学物理

题目描述

为了测试小W的物理水平，Mr.X在二维坐标系中放了N面镜子(镜子坐标绝对值不超过M)，镜子均与坐标轴成45°角，所以一共有两种类型“/”和“\”。原点不会有镜子，任意一点最多只有一面镜子。
镜子两个面都能反光，而中间不透光，例如，对于一个“/”型镜子，下方向射入的光线会被反射到右方向，左方向射入的光线会被反射到上方向。
现在有一条光线从原点沿X轴正方向射出，求走过T路程后所在位置

模拟

就模拟它做
可能有循环节，找到了循环节就把步数模掉

#include<cstdio> 
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
int up[maxn],down[maxn],left[maxn],right[maxn],wz[maxn];
struct dong{
    int x,y,id;
    bool p;/* 0 / 1 \ */
} a[maxn];
bool bz[maxn][4];//0 up 1 down 2 left 3 right
char ch;
int i,j,k,l,r,t,n,m,xx,yy;
ll p,len;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
char get(){
    char ch=getchar();
    while (ch==' ')  ch=getchar();
    return ch;
}
bool cmp1(dong a,dong b){
    return (a.x<b.x)||(a.x==b.x&&a.y<b.y);
}
bool cmp2(dong a,dong b){
    return (a.y<b.y)||(a.y==b.y&&a.x<b.x);
}
int getdis(int j,int k){
    return abs(a[j].x-a[k].x)+abs(a[j].y-a[k].y);
}
void work(int x,int y,int r,ll p){
    if (r==0) printf("%lld %lld\n",(ll)x,(ll)y+p);
    else if (r==1) printf("%lld %lld\n",(ll)x,(ll)y-p);
    else if (r==2) printf("%lld %lld\n",(ll)x-p,(ll)y);
    else printf("%lld %lld\n",(ll)x+p,(ll)y);
}
int main(){
    freopen("mir.in","r",stdin);freopen("mir.out","w",stdout);
    n=read();m=read();scanf("%lld",&p);
    fo(i,1,n){
        a[i].x=read();a[i].y=read();
        ch=getchar();
        if (ch!='/') a[i].p=1;
        a[i].id=i;
    }
    sort(a+1,a+n+1,cmp1);
    fo(i,1,n){
        if (i>1&&a[i-1].x==a[i].x) down[a[i].id]=a[i-1].id;
        if (i<n&&a[i+1].x==a[i].x) up[a[i].id]=a[i+1].id;
    }
    sort(a+1,a+n+1,cmp2);
    fo(i,1,n){
        if (i>1&&a[i-1].y==a[i].y) left[a[i].id]=a[i-1].id;
        if (i<n&&a[i+1].y==a[i].y) right[a[i].id]=a[i+1].id;
    }
    fo(i,1,n) wz[a[i].id]=i;
    t=0;
    fo(i,1,n)
        if (a[i].x>0&&a[i].y==0&&(!t||a[i].x<a[t].x)) t=i;
    if (!t||p<=a[t].x){
        work(0,0,3,p);
        return 0;
    }
    p-=(ll)a[t].x;
    bz[a[t].id][3]=1;
    r=3;
    while (1){
        if (a[t].p){
            if (r==0) l=2;
            if (r==1) l=3;
            if (r==2) l=0;
            if (r==3) l=1;
        }
        else{
            if (r==0) l=3;
            if (r==1) l=2;
            if (r==2) l=1;
            if (r==3) l=0;
        }
        if (l==0) k=up[a[t].id];
        if (l==1) k=down[a[t].id];
        if (l==2) k=left[a[t].id];
        if (l==3) k=right[a[t].id];
        k=wz[k];
        if (!k||p<=getdis(t,k)){
            work(a[t].x,a[t].y,l,p);
            return 0;
        }
        p-=(ll)getdis(t,k);
        if (bz[a[k].id][l]){
            t=k;
            r=l;
            break;
        }
        bz[a[k].id][l]=1;
        t=k;
        r=l;
    }
    xx=t;yy=r;
    while (1){
        if (a[t].p){
            if (r==0) l=2;
            if (r==1) l=3;
            if (r==2) l=0;
            if (r==3) l=1;
        }
        else{
            if (r==0) l=3;
            if (r==1) l=2;
            if (r==2) l=1;
            if (r==3) l=0;
        }
        if (l==0) k=up[a[t].id];
        if (l==1) k=down[a[t].id];
        if (l==2) k=left[a[t].id];
        if (l==3) k=right[a[t].id];
        k=wz[k];
        if (!k||p<=getdis(t,k)){
            work(a[t].x,a[t].y,l,p);
            return 0;
        }
        p-=(ll)getdis(t,k);
        len+=(ll)getdis(t,k);
        if (k==xx&&l==yy){
            t=k;
            r=l;
            break;
        }
        t=k;
        r=l;
    }
    p%=len;
    if (!p){
        printf("%d %d\n",a[xx].x,a[xx].y);
        return 0;
    }
    while (1){
        if (a[t].p){
            if (r==0) l=2;
            if (r==1) l=3;
            if (r==2) l=0;
            if (r==3) l=1;
        }
        else{
            if (r==0) l=3;
            if (r==1) l=2;
            if (r==2) l=1;
            if (r==3) l=0;
        }
        if (l==0) k=up[a[t].id];
        if (l==1) k=down[a[t].id];
        if (l==2) k=left[a[t].id];
        if (l==3) k=right[a[t].id];
        k=wz[k];
        if (!k||p<=getdis(t,k)){
            work(a[t].x,a[t].y,l,p);
            return 0;
        }
        p-=(ll)getdis(t,k);
        t=k;
        r=l;
    }
}