NYOJ 511 - 移动小球 链表版

#include <stdio.h>
struct ball
{
int num;
struct ball *left, *right;
}ball[11000];
void Initial(int n);
void Leftmove(int x, int y);
void Rightmove(int x, int y);
int main()
{
//freopen("input.txt","r",stdin);
int T;
scanf("%d",&T);
while (T--)
{
int n, m;
char c;
scanf("%d%d", &n, &m);
Initial(n);
while (m--)
{
char c;
int x, y;
getchar();
scanf("%c%d%d",&c, &x, &y);
if (c == 'A')
Leftmove(x, y);
else if (c == 'B')
Rightmove(x, y);
else
if (x == 1)
printf("%d\n",ball[y].right->num);
else
printf("%d\n",ball[y].left->num);
}
}
return 0;
}
void Initial(int n)
{
for (int i = 1; i <= n; i++)
{
ball[i].num = i;
ball[i].left = &ball[i - 1];
ball[i].right = &ball[i + 1];
}
ball[1].left = &ball
;
ball
.right = &ball[1];
}
void Leftmove(int x, int y)
{
ball[x].left->right = ball[x].right;
ball[x].right->left = ball[x].left;
ball[x].right = &ball[y];
ball[x].left = ball[y].left;
ball[x].left->right = &ball[x];
ball[y].left = &ball[x];
}
void Rightmove(int x, int y)
{
ball[x].left->right = ball[x].right;
ball[x].right->left = ball[x].left;
ball[y].right->left = &ball[x];
ball[x].right = ball[y].right;
ball[x].left = &ball[y];
ball[y].right = &ball[x];
}

#include <bits/stdc++.h>
using namespace std;

int main()
{
//freopen("input.txt", "r", stdin);
list<int> ball;
list<int>::iterator itNow, itTarget;
int T, n, i, num, now, target;
char ch;
scanf("%d", &T);
while (T--)
{
ball.clear();
scanf("%d%d%*c", &num, &n);
for (i = 1; i <= num; i++)
ball.push_back(i);
while (n--)
{
ch = getchar();
scanf("%d%d%*c", &now, &target);
itNow = find(ball.begin(), ball.end(), now);
itTarget = find(ball.begin(), ball.end(), target);
if (ch == 'A')
{
ball.insert(itTarget, now);
ball.erase(itNow);
}
else if (ch == 'B')
{
ball.insert(++itTarget, now);
ball.erase(itNow);
}
else		//1询问右边的, 0左边.
{
if (now == 1 && itTarget == --ball.end())
printf("%d\n", *ball.begin());
else if (now == 0 && itTarget == ball.begin())
printf("%d\n", *(--ball.end()));
else if (now == 1)
printf("%d\n", *(++itTarget));
else if (now == 0)
printf("%d\n", *(--itTarget));
}
}
}
return 0;
}