2014百度之星初赛第一轮解题报告：information

#include <set>
#include <string>
#include <stdio.h>
using namespace std;

struct Tank {
int id;
int x;
int y;
};

Tank tank[100005];
int id;
set<int> tanks[100005];
FILE* fout = NULL;

void insert(int m, int a, int b, int x, int y) {
if (m == 1) {
if (tank[b].id == -2) {
tank[b].id = id++;
tank[b].x = tank[b].y = 0;
tanks[tank[b].id].insert(b);
}
if (tank[a].id >= 0) {
tanks[tank[a].id].erase(a);
}
tank[a].id = tank[b].id;
if (tank[a].id >= 0) {
tanks[tank[a].id].insert(a);
}
tank[a].x = tank[b].x + x;
tank[a].y = tank[b].y + y;
}
else if (m == 2) {
if (tank[a].id >= 0) {
tanks[tank[a].id].erase(a);
}
tank[a].id = -1;
tank[a].x = x;
tank[a].y = y;
}
else if (m == 3) {
if (tank[a].id == -2) {
insert(1, a, b, x, y);
}
else if (tank[b].id == -2) {
insert(1, b, a, -x, -y);
}
else if (tank[a].id == -1 && tank[b].id == -1) {
if (tank[a].x != tank[b].x + x || tank[a].y != tank[b].y + y) {
fprintf(fout, "REJECT\n");
}
}
else if (tank[a].id == -1) {
insert(4, b, -1, tank[a].x - x, tank[a].y - y);
}
else if (tank[b].id == -1) {
insert(4, a, -1, tank[b].x + x, tank[b].y + y);
}
else {
if (tank[a].id == tank[b].id) {
if (tank[a].x != tank[b].x + x || tank[a].y != tank[b].y + y) {
fprintf(fout, "REJECT\n");
}
}
else {
int dx = tank[b].x + x - tank[a].x;
int dy = tank[b].y + y - tank[a].y;
int j = tank[a].id;
int k = tank[b].id;
for (set<int>::const_iterator itr = tanks[j].begin(); itr != tanks[j].end(); itr++) {
tank[*itr].id = tank[b].id;
tank[*itr].x += dx;
tank[*itr].y += dy;
tanks[k].insert(*itr);
}
set<int> temp;
tanks[j].swap(temp);
}
}
}
else if (m == 4) {
if (tank[a].id == -1) {
if (tank[a].x != x || tank[a].y != y) {
fprintf(fout, "REJECT\n");
}
}
else if (tank[a].id == -2) {
tank[a].id = -1;
tank[a].x = x;
tank[a].y = y;
}
else {
int dx = x - tank[a].x;
int dy = y - tank[a].y;
int j = tank[a].id;
for (set<int>::const_iterator itr = tanks[j].begin(); itr != tanks[j].end(); itr++) {
tank[*itr].id = -1;
tank[*itr].x += dx;
tank[*itr].y += dy;
}
set<int> temp;
tanks[j].swap(temp);
}
}
}

void query(int m, int a, int b) {
if (m == 5) {
if (tank[a].id > -2 && tank[b].id > -2 && tank[a].id == tank[b].id) {
fprintf(fout, "%d %d\n", tank[a].x - tank[b].x, tank[a].y - tank[b].y);
}
else {
fprintf(fout, "UNKNOWN\n", m, a, b);
}
}
else {
if (tank[a].id == -1) {
fprintf(fout, "%d %d\n", tank[a].x, tank[a].y);
}
else {
fprintf(fout, "UNKNOWN\n", m, a);
}
}
}

int main() {
int T, N, t = 1;
int m = 0, a = 0, b = 0, x = 0, y = 0;
FILE* fp = stdin; //fopen("data.in", "r");
fout = stdout; //fopen("output", "w");

fscanf(fp, "%d", &T);
while (T--) {
fscanf(fp, "%d", &N);
for (int i = 0; i <= N; i++) {
tank[i].id = -2;
}
id = 0;
fprintf(fout, "Case #%d:\n", t++);

while (N--) {
fscanf(fp, "%d", &m);
if (m == 1 || m == 3) {
fscanf(fp, "%d %d %d %d", &a, &b, &x, &y);
insert(m, a, b, x, y);
}
else if (m == 2 || m == 4) {
fscanf(fp, "%d %d %d", &a, &x, &y);
insert(m, a, b, x, y);
}
else if (m == 5) {
fscanf(fp, "%d %d", &a, &b);
query(m, a, b);
}
else if (m == 6) {
fscanf(fp, "%d", &a);
query(m, a, b);
}
}

for (int i = 0; i < id; i++) {
set<int> temp;
tanks[i].swap(temp);
}
}
}