题目大意,
给定一个有n个点的环,n不超过10,每个点上有一个权重
起始时权重将会给出,然后有2种操作
第一种操作是,选择一个位置i,获得权重w = a[i],把a[i]变成0,然后接下来在环上顺着走,每个数都加1,直到w为0
第二种操作是,选择一个位置i,在换上倒着走,每个数都减去1,然后收集这个权重w,直到遇见0,然后把w填到0里
求一种方案,使得从起始的权重到达终止的权重
不难发现,这两种操作互为逆操作
所以只用第一种操作
先把起始权重全部集中到a[0]
终止权重也全部集中到a[0]
然后最终方案就是方案一加方案二的逆操作
orz
#include <iostream> #include <cstdio> #include <vector> #include <algorithm> using namespace std; typedef vector<int> VI; VI solve(VI a){ int n = a.size(); int sum = 0; VI ans; for(auto x : a) sum += x; while(a[0] != sum){ for(int i = 1; i < n; i++){ while(a[i] != 0){ ans.push_back(i); int p = (i+1)%n, t = a[i]; a[i] = 0; while(t){ a[p]++; t--; p = (p+1)%n; } } } } return ans; } class ReverseMancala{ public: VI findMoves(VI a, VI b){ VI x = solve(a); VI y = solve(b); VI z; int n = a.size(); reverse(y.begin(), y.end()); for(auto X : x) z.push_back(X); for(auto X : y) z.push_back(X+n); return z; } };
时间: 2024-11-08 21:50:49