算典05_例题_02_UVA-101

最新推荐文章于 2018-08-13 12:00:24 发布

_duadua

最新推荐文章于 2018-08-13 12:00:24 发布

阅读量295

点赞数

分类专栏：算法竞赛入门经典(紫书)

本文链接：https://blog.csdn.net/a27038/article/details/68947076

版权

算法竞赛入门经典(紫书) 专栏收录该内容

71 篇文章 0 订阅

订阅专栏

The Blocks Problem

题意

有一排石子从0到n-1编号，有四种操作来移动石子:

1.move a onto b
  将石子a移到石子b上，在此之前将所有在a上面的石子移
  回它最初的位置，将所有石子b上的石子也移回它最初的
  位置
2.move a over b
  将石子a上面的石子全部移回它们最初的位置，然后将石
  子a放到石子b所在堆的最上面
3.pile a onto b
  将石子b上面的石子全部移回它们最初的位置，然后将石
  子a和它上面的所有石子按当前顺序移到b的上面
4.pile a over b
  将石子a和它上面的所有石子按当前顺序移到石子b所在
  堆的最上面

输出最后各堆的石子情况

题解

使用vector < int > ans[maxn] 保存石子的分布情况
使用一个数组cur[maxn]保存石子i当前所在的堆的编号
然后模拟以上四种操作即可
只要用vector的push_back()，pop_back()和resize()即可

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <vector>

using namespace std;
const int maxn = 1e2 + 5;
const int INF = (1<<31)-1;
#define met(a, b) memset(a, b, sizeof(a));
#define IN freopen("in.txt", "r", stdin);
typedef long long LL;

int n, a, b, cur[50];
char s[50], o1[10], o2[10];
vector<int> ans[maxn];

bool read() {
    fgets(s, 50, stdin);
    if(s[0] == 'q') return false;
    return true;
}

void solve() {
    if(o1[0] == 'm') {
        int id = cur[a], id2 = cur[b];
        int len = ans[id].size()-1;
        for(int i = len; i >= 0 && ans[id][i] != a; --i) {
            ans[ans[id][i]].push_back(ans[id][i]);
            cur[ans[id][i]] = ans[id][i];
            ans[id].pop_back();
        }
        if(o2[3] == 'o') {

            len = ans[id2].size()-1;
            for(int i = len; i >= 0 && ans[id2][i] != b; --i) {
                ans[ans[id2][i]].push_back(ans[id2][i]);
                cur[ans[id2][i]] = ans[id2][i];
                ans[id2].pop_back();
            }
        }
        ans[id].pop_back(); ans[id2].push_back(a); cur[a] = id2;

    } else if(o1[0] == 'p') {
        int id2 = cur[b];
        int len = ans[id2].size()-1;
        if(o2[3] == 'o') {
            for(int i = len; i >= 0 && ans[id2][i] != b; --i) {
                ans[ans[id2][i]].push_back(ans[id2][i]);
                cur[ans[id2][i]] = ans[id2][i];
                ans[id2].pop_back();
            }
        }
        int id = cur[a], idx = 0;
        len = ans[id].size()-1;
        while(idx <= len && ans[id][idx] != a) ++idx;
        for(int i = idx; i <= len; ++i) {
            ans[id2].push_back(ans[id][i]);
            cur[ans[id][i]] = id2;
        }
        ans[id].resize(idx);
    }
}

int main(){
    #ifdef _LOCAL
    IN;
    #endif // _LOCAL

    scanf("%d", &n);getchar();
    for(int i = 0; i < n; ++i) cur[i] = i, ans[i].push_back(i);
    while(read()) {
        sscanf(s, "%s%d%s%d", o1, &a, o2, &b);
        if(cur[a] == cur[b])continue;
        solve();
    }

    for(int i = 0; i < n; ++i) {
        printf("%d:", i);
        for(int j = 0; j < ans[i].size(); ++j) printf(" %d", ans[i][j]);
        printf("\n");
    }
    return 0;
}