忘记贴个中文题面了~~
作为前来上海大学参加ICPC比赛的退役ACMer现役JBer,小明每次来上海大学参加比赛,都会打铁。心里就会产生一种"这是什么JB比赛"的念头。为了排解痛苦,他就会去上海的迪士尼坐过山车。他很喜欢这种血压拉满的感觉,让他感觉自己如同变成了一只"快乐"的Flappy Bird,忘记了所有的WA和TLE……
过山车有一系列的高低起伏的转折点,各个转折点的高度,按照路径顺序形成了一个数列a[],小明坐完所有过山车后,他的血压可以认为是Σ(|a[i + 1] - a[i]|),即相邻过山车高度的绝对值之和。
小明比赛一直打铁,一直打铁,于是一直需要不停地坐过山车,然而,可以使得他忘记烦恼的所需血压阈值也在慢慢上升。渐渐地,上海迪士尼的过山车,已经达不到小明对自己血压的要求了。
因此,小明开始设想增加额外的m个转折点b[],以任意的位置和顺序(此处需要讨论具体规则),加入原有的过山车的路径a[]中,让自己的血压拉得尽可能高。
请计算一下,小明的血压最高可以变成多少呢?这非常重要。我们需要这个数据来请一位适合的心血管内科医生。救救小明吧!您的AC非常重要!
再反手给一个知乎的原问题链接 ——
https://www.zhihu.com/question/355256075/answer/907869387
当然,事情前前后后的链接可多了2333 ~
以下内容的发表时间是 —— 2019年11月25日 20:30:07
血压游戏,实质名归好吧233333~
代码时间是 2019-11-23 18:34:27一个标点符号我都没有修改,贴在这里了。
询问了出题组包括bin巨,他们同意我把标程发过来。
不排除有错误,我觉得很有可能会被发现哪里写得有问题,但是能学到知识不是美滋滋吗 ~
而且确实,在我看来,就与我写了更优复杂度的标程没什么区别,除了n的设置外,没有影响比赛结果。问心无愧。
// #include <bits/stdc++.h>
#include<stdio.h>
#include<iostream>
#include<string>
#include<string.h>
#include<set>
#include<map>
#include<vector>
#include<queue>
#include<algorithm>
#include<time.h>
using namespace std;
#define FMS(x, y, g) memset(x, y, sizeof(x[0]) * (g))
typedef long long LL;
// Variables For the Specific Problem
const bool GUESS = false; // 是否验证一些猜想(如点数、边数等)的开关
const bool DEBUG = false; // 是否输出细节(用于调试)的开关
// control variables end
const LL INF = 1e16; // 表示两点之前的极大收益 [1E9是不够的哦]
const int E9 = 1e9; // 就表示 1E9
const int G = 100000; // 题目设置的数组长度
const int N = G * 4 + 99; // 点数 4n 级别
const int M = G * 10 * 2 + 99; // 边数 10n 级别
int n, m; // n 表示原始数组 a 的大小,m 表示可插入集合 b 的大小
int a[N], b[N]; // 待插入数组 a, 可插入集合 b
int casenum, casei; // 数据组数 casenum, 当前数据编号 casei
// End
// 把 a 或 b 的信息整合一起的数据类型
struct Ele
{
int tp, id;
int l, r;
Ele(){};
Ele(int tp_, int id_, int l_, int r_) {
tp = tp_; id = id_; l = l_; r = r_;
};
};
// 按照左界下降排序
bool cmpL(Ele a, Ele b) {
if(a.l != b.l)return a.l > b.l;
return a.tp > b.tp;
}
// 按照右界上升排序
bool cmpR(Ele a, Ele b) {
if(a.r != b.r)return a.r < b.r;
return a.tp > b.tp;
}
// 费用流
struct wkcMCMF {
int ST, ED; // 源点与汇点(我习惯上使得ST=0,ED=最后一个点的编号)
int first[N], ID; // 边集的起点边编号(ID初始化为1,表示新分配的边的编号)
int w[M], cap[M], cost[M], nxt[M]; // 边包括了(抵达点w,容量cap,单位流量成本cost,下条边编号nxt)等信息
LL f[N]; // f[x]表示在残量网络下,从源点到达x的最小距离
int pe[N]; // pe[x]记录流向x的前驱边
bool e[N]; // e[x]是判定点x是否在SPFA队列中的辅助数组
queue<int> q; // q是SPFA的队列
int SUM; // 用于检查程序正确性——求最终血压值
// 加边
void ins(int x, int y, int cap_, LL cost_) {
w[++ID] = y;
cap[ID] = cap_;
cost[ID] = cost_;
nxt[ID] = first[x];
first[x] = ID;
w[++ID] = x;
cap[ID] = 0;
cost[ID] = -cost_;
nxt[ID] = first[y];
first[y] = ID;
}
// 入队
void inq(int x, LL cost_, int pe_) {
if (cost_ <= f[x])return; // 单位流量收益更小,没有更新意义
f[x] = cost_; // 单位流量收益大的条件下做更新
pe[x] = pe_; // 然后记录上一条边
if (x == ED || e[x])return; // SPFA的入队标记以防止重复入队做冗余更新
e[x] = true;
q.push(x);
}
// SPFA找最长路(最高收益)
int Augmenting = 1; // 表示具有增广意义的最低增广收益值,可能会有调整为 1 或者 0 的需要
bool spfa() { // 返回值为true表示找到了一条成功的增广路
FMS(f, -63, ED + 2); // 初始的收益值一定要设置为极小 [-1似乎是不够的]
cap[0] = E9;
inq(ST, 0, 0);
while (!q.empty()) {
int x = q.front();
q.pop();
e[x] = 0;
for (int z = first[x]; z; z = nxt[z]) {
if (cap[z])inq(w[z], f[x] + cost[z], z);
}
}
return f[ED] >= Augmenting; // 收益 < Augmenting 之后的增广便不再有意义了
}
// 从终点滚到起点,确定此费用下的最大流量,并修改残余流量 [单路回溯的普通写法]
vector<LL>slowMCMF(LL basic) {
vector<LL>rtn;
int maxflow = 0;
LL mincost = basic;
while (spfa()) {
int flow = E9;
int x = ED;
while (x != ST) {
flow = min(flow, cap[pe[x]]);
x = w[pe[x] ^ 1];
}
maxflow += flow;
x = ED;
while (x != ST) {
cap[pe[x]] -= flow;
cap[pe[x] ^ 1] += flow;
x = w[pe[x] ^ 1];
}
for(int i = 1; i <= flow; ++i) {
mincost += f[ED];
rtn.push_back(mincost);
}
}
while(rtn.size() < m) {
rtn.push_back(mincost);
}
return rtn;
}
// 高效费用流所使用的DFS多路回溯算法
bool vis[N];
int dfs(int x, int all) {
if (x == ST)return all;
int use = 0;
vis[x] = true;
for (int z = first[x]; z; z = nxt[z])
if (cap[z ^ 1]) {
int y = w[z];
if (!vis[y] && f[y] + cost[z ^ 1] == f[x]) {
int tmp = dfs(y, min(cap[z ^ 1], all - use));
cap[z ^ 1] -= tmp;
cap[z] += tmp;
use += tmp;
if (use == all)break;
}
}
return use;
}
// 从终点滚到起点,确定此费用下的最大流量,并修改残余流量 [多路回溯的DFS高效写法]
vector<LL> fastMCMF(LL basic) {
vector<LL>rtn;
int maxflow = 0;
LL mincost = basic;
while (spfa()) {
int flow;
while (FMS(vis, 0, ED + 2),
flow = dfs(ED, E9)) {
maxflow += flow;
for(int i = 1; i <= flow; ++i) {
mincost += f[ED];
rtn.push_back(mincost);
}
}
}
while(rtn.size() < m) {
rtn.push_back(mincost);
}
return rtn;
}
// ST = 0
// added point [1, m]
// up segment point [m + 1, m + n)
// down segment point [m + n + 1, m + n + n)
// final [m + n + n + 0, m + n + n + n]
// ED = m + n * 3 + 1
Ele ele[N];
map<int, int>rkL; // 记录每个左界区间对应的排名(start from 1)
map<int, int>rkR; // 记录每个右界区间对应的排名(start from 1)
pair<int, int>vaL[N]; // sorted L list (vaL, id) down
pair<int, int>vaR[N]; // sorted R list (vaR, id) up
int pos[N]; // a to b, record the ans
vector<int>ansVec; // ans vector
void NplusM_mapBuild() {
ID = 1;
ST = 0;
ED = m + n * 3 + 1;
int eg = 0;
for (int i = 1; i <= m; ++i) {
ele[++eg].tp = 1;
ele[eg].id = i;
ele[eg].l = ele[eg].r = b[i];
}
for (int i = 1; i < n; ++i) {
ele[++eg].tp = 2;
ele[eg].id = i;
ele[eg].l = min(a[i], a[i + 1]);
ele[eg].r = max(a[i], a[i + 1]);
}
rkL.clear();
rkR.clear();
// point -> bigger segment [L decreasing order]
// 从区间 [l[i + 1], r[i + 1]] 向区间 [l[i], r[i]] 连一条容量极大(>=m即可),收益为(l[i] - l[i + 1]) * 2的边 <此处n - 2条边>
int lastId = E9;
int lastV;
int oL = 0;
sort(ele + 1, ele + eg + 1, cmpL);
for(int i = 1; i <= eg; ++i) {
if(ele[i].tp == 2) {
int eid = ele[i].id + m;
if(lastId != E9) {
ins(eid, lastId, E9, (lastV - ele[i].l) << 1);
}
lastId = eid;
lastV = ele[i].l;
vaL[++oL] = {lastV, ele[i].id};
rkL[lastV] = oL;
}
else if(lastId != E9) {
// 向比其大的第一个(如果存在) "下匹配的左界区间[l, r]", 连一条容量为1(>=1即可),收益为(l - v) * 2的边
ins(ele[i].id, lastId, 1, (lastV - ele[i].l) << 1);
}
}
if (DEBUG) {
printf("rkL(v,g): ");
for(auto &it : rkL)printf("[%d %d] ", it.first, it.second);
puts("");
}
// point -> smaller segment [R increasing order]
// 从区间 [l[i + 1], r[i + 1]] 向区间 [l[i], r[i]] 连一条容量极大(>=m即可),收益为(r[i + 1] - r[i]) * 2的边 <此处n - 2条边>
lastId = E9;
int oR = 0;
sort(ele + 1, ele + eg + 1, cmpR);
for (int i = 1; i <= eg; ++i) {
if (ele[i].tp == 2) {
int eid = ele[i].id + m + n;
if(lastId != E9) {
ins(eid, lastId, E9, (ele[i].r - lastV) << 1);
}
lastId = eid;
lastV = ele[i].r;
vaR[++oR] = {lastV, ele[i].id};
rkR[lastV] = oR;
}
else if (lastId != E9) {
// 向比其大的第一个(如果存在) "下匹配的左界区间[l, r]", 连一条容量为1(>=1即可),收益为(v - r) * 2的边
ins(ele[i].id, lastId, 1, (ele[i].r - lastV) << 1);
}
}
if (DEBUG) {
printf("rkR(v,g): ");
for(auto &it : rkR)printf("[%d %d] ", it.first, it.second);
puts("");
}
// 从 m 个插入点向首尾 2 个特殊插入位置连一条容量为1(>=1即可),收益为0的边 <此处m * 2条边>
for (int i = 1; i <= m; ++i) {
ins(i, m + n + n + 0, 1, abs(b[i] - a[1]));
ins(i, m + n + n + n, 1, abs(b[i] - a[n]));
}
// 源点连接 m 个被插入点[1, m], 容量统一为1,收益统一为0 <此处m条边>
for (int i = 1; i <= m; ++i) {
ins(ST, i, 1, 0);
}
// 下匹配左区间点 & 上匹配右区间点 -> 真实区间点 <此处(n-1)*2条边>
for (int i = 1; i < n; ++i) {
ins(m + i, m + n + n + i, 1, 0);
ins(m + n + i, m + n + n + i, 1, 0);
}
// 真实区间点 [m + n + n + 0, m + n + n + n] 向汇点 ED 连一条容量为1,收益为0的边 <此处n + 1条边>
for (int i = 0; i <= n; ++i) {
ins(m + n + n + i, ED, 1, 0);
}
}
// 根据网络流的流量情况构造解
int from[N]; // 记录每个实际的区间a是作为上升区间还是下降区间被匹配的
void NplusM_output() {
set<int>AnyPosOK_bset;
for (int i = 1; i <= m; ++i) {
AnyPosOK_bset.insert(b[i]);
}
vector<Ele>LbTOa;
vector<Ele>RbTOa;
for (int y = 0; y <= n; ++y) {
// check[m + 1, m + n) && [m + n + 1, m + n + n)
if (y != 0 && y != n) {
int va = min(a[y], a[y + 1]);
for (int z = first[m + y]; z; z = nxt[z]) {
if (z % 2 == 1 && cap[z] != 0 && w[z] <= m) {
int vb = b[w[z]];
int va = min(a[y], a[y + 1]);
AnyPosOK_bset.erase(vb);
LbTOa.push_back({vb, rkL[va], va, va});
if (DEBUG) {
printf("LbTOa: (b = %d a = %d) va = %d vb = %d\n", w[z], y, va, vb);
}
}
}
va = max(a[y], a[y + 1]);
for (int z = first[m + n + y]; z; z = nxt[z]) {
if (z % 2 == 1 && cap[z] != 0 && w[z] <= m) {
int vb = b[w[z]];
AnyPosOK_bset.erase(vb);
RbTOa.push_back({vb, rkR[va], va, va});
if (DEBUG) {
printf("RbTOa: (b = %d a = %d) va = %d vb = %d\n", w[z], y, va, vb);
}
}
}
// 查询每个区间实际是匹配的上升区间还是下降区间
from[y] = 0;
for (int z = first[m + n + n + y]; z; z = nxt[z]) {
if (z % 2 == 1 && cap[z] != 0) {
if (w[z] > m && w[z] <= m + n) {
from[y] = 1;
} else if (w[z] > m + n && w[z] <= m + n + n) {
from[y] = 2;
}
}
}
}
// 这里其实只有 y == 0 或 y == n 时才会被直接从[1, m]产生流量 [TODO——验证猜想]
for (int z = first[m + n + n + y]; z; z = nxt[z]) {
if (z % 2 == 1 && cap[z] != 0 && w[z] <= m) {
int vb = b[w[z]];
AnyPosOK_bset.erase(vb);
break;
}
}
}
// In LbTOa or RbTOa, {tp:vb, id:rk, l:va, r:va}, the same value makes rk seem to be bigger
// In vaL or vaR, {vL or vR, id}]
// vaL[i], the i-th smallest val and pos
// vaR[i], the i-th biggest val and pos
FMS(pos, 0, n + 2);
int LpreID = 0;
sort(LbTOa.begin(), LbTOa.end(), cmpL); // vaL is going down after sorting
for(auto &it : LbTOa) {
LpreID = min(it.id, LpreID + 1);
while(true) {
int p = vaL[LpreID].second;
if (from[p] != 1)++LpreID;
else {
pos[p] = it.tp;
break;
}
}
}
int RpreID = 0;
sort(RbTOa.begin(), RbTOa.end(), cmpR); // vaR is going up after sorting
for(auto &it : RbTOa) {
it.id = min(it.id, RpreID + 1);
RpreID = it.id;
while(true) {
int p = vaR[RpreID].second;
if (from[p] != 2)++RpreID;
else {
pos[p] = it.tp;
break;
}
}
}
if (DEBUG) {
printf("POS: ");
for(int i = 1; i < n; ++i)printf("%d ", pos[i]);
puts("");
}
// ans output
ansVec.clear();
for (int y = 0; y <= n; ++y) {
if (y) {
ansVec.push_back(a[y]);
}
// possibility 1: matched already
if (y != 0 && y != n && pos[y]) {
ansVec.push_back(pos[y]);
continue;
}
// possibility 2: head or tail
bool flag = 0;
if (y == 0 || y == n) {
for (int z = first[m + n + n + y]; z; z = nxt[z]){
if (z % 2 == 1 && cap[z] != 0 && w[z] <= m) {
ansVec.push_back(b[w[z]]);
flag = 1;
break;
}
}
}
// possibility 3: any position is the same to some elements
if(!flag && AnyPosOK_bset.size()) {
ansVec.push_back(*AnyPosOK_bset.begin());
AnyPosOK_bset.erase(AnyPosOK_bset.begin());
}
}
SUM = 0;
for(int i = 0; i < ansVec.size(); ++i) {
printf("%d ", ansVec[i]);
if (i)SUM += abs(ansVec[i] - ansVec[i - 1]);
}puts("");
}
// 低效建图法,边数O(nm)
void NmultM_mapBuild() {
ID = 1;
ST = 0;
ED = m + 1 + n + 1;
for (int i = 1; i <= m; ++i) {
int V = b[i];
ins(ST, i, 1, 0);
for (int j = 0; j <= n; ++j) {
int lftV = j == 0 ? V : a[j];
int rgtV = j == n ? V : a[j + 1];
int oriV = (j == 0 || j == n) ? 0 : abs(a[j + 1] - a[j]);
int incV = abs(lftV - V) + abs(rgtV - V) - oriV;
ins(i, m + 1 + j, 1, incV);
}
}
for (int i = 0; i <= n; ++i) {
ins(m + 1 + i, ED, 1, 0);
}
}
// 低效建图下的解构造
void NmultM_output() {
// ST = 0
// b[] ∈ [1, m]
// a[] ∈ [m + 1 + 0, m + 1 + n]
// ED = n + m + 2
vector<int>ansVec;
for (int i = 0; i <= n; ++i) {
pos[i] = 0;
}
for (int x = 1; x <= m; ++x) {
for (int z = first[x]; z; z = nxt[z]) {
if (z % 2 == 0 && cap[z] == 0) {
pos[w[z] - m - 1] = b[x];
}
}
}
for (int i = 0; i <= n; ++i) {
if(i) ansVec.push_back(a[i]);
if(pos[i])ansVec.push_back(pos[i]);
}
for(auto &it : ansVec) {
printf("%d ", it);
}puts("");
}
// 输出实际的流量网络图的DEBUG过程
void printMap() {
for (int x = 0; x <= ED; ++x) {
for (int z = first[x]; z; z = nxt[z]) {
if (z % 2 == 0) {
int y = w[z];
if (cap[z ^ 1] != 0)
printf("%d->%d(%d, %d)\n", x, y, cap[z ^ 1], cost[z]);
}
}
}
}
}mcmf;
//暴力DFS算法
struct My_BF {
int rev[1 << 20];
LL MAXV;
vector<LL>BEST;
vector<int>ansVec, vec;
void init() {
for(int i = 0; i < 20; ++i) {
rev[1 << i] = i;
}
}
int lowbit(int x) {
return x & -x;
}
void dfs(int mask, int pos, LL sumV, int num) {
if (num == m && sumV > MAXV) {
MAXV = sumV;
ansVec = vec;
}
if (sumV > BEST[num - 1]) {
BEST[num - 1] = sumV;
}
if (pos > n) {
return;
}
for (int tmp = mask; tmp; tmp -= lowbit(tmp)) {
int o = rev[lowbit(tmp)];
int v = b[o + 1];
int addV = 0;
if (pos == 0) {
addV = abs(v - a[pos + 1]);
}
else if (pos == n) {
addV = abs(v - a[pos]);
}
else {
addV = abs(v - a[pos + 1]) + abs(v - a[pos]);
}
vec.push_back(v); if (pos < n)vec.push_back(a[pos + 1]);
dfs(mask - lowbit(tmp), pos + 1, sumV + addV, num + 1);
vec.pop_back(); if (pos < n)vec.pop_back();
}
int addV = 0;
if (pos != 0 && pos != n) {
addV = abs(a[pos] - a[pos + 1]);
}
if (pos < n)vec.push_back(a[pos + 1]);
dfs(mask, pos + 1, sumV + addV, num);
if (pos < n)vec.pop_back();
}
vector<LL> solve() {
init();
BEST.resize(m);
for (int i = 0; i <= m; ++i) {
BEST[i] = -1;
}
MAXV = -1;
dfs((1 << m) - 1, 0, 0, 0);
return BEST;
}
void output() {
for (int i = 0; i < ansVec.size(); ++i) {
printf("%d ", ansVec[i]);
}
puts("");
}
}bf;
// 贪心算法——TODO
struct My_Greedy {
LL solve() {
return 0;
}
}greedy;
// 数据生成器
struct My_DataGenerator {
void rd_dataGenerator() {
srand(time(0));
freopen("Blood Pressure Game.in", "w", stdout);
casenum = 1000; printf("%d\n", casenum + 0);
int smlCasenum = 990;
int midDCasenum = 997;
for (casei = 1; casei <= casenum; ++casei)
{
int n = rand() % 100 + 1;
if (casei > smlCasenum) {
n = rand() % 600 + 1;
}
else if(casei > midDCasenum) {
n = 600;
}
m = rand() % (n + 1) + 1;
int TOPV = casei <= smlCasenum ? n * 10 : (rand() % 2 ? 10000 : E9);
printf("%d %d\n", n, m);
set<int>noEqualSot;
for(int i = 1; i <= n; ++i) {
do
{
a[i] = rand() % TOPV + 1;
}while(noEqualSot.count(a[i]));
noEqualSot.insert(a[i]);
printf("%d ", a[i]);
}puts("");
for(int i = 1; i <= m; ++i) {
do
{
b[i] = rand() % TOPV + 1;
}while(noEqualSot.count(b[i]));
noEqualSot.insert(b[i]);
printf("%d ", b[i]);
}puts("");
}
}
}dataGenerator;
struct Checker {
string check_it() {
auto& ansVec = mcmf.ansVec;
set<int>sot;
for (int i = 0; i < n + m; ++i) {
int v = ansVec[i];
if(sot.count(v)) {
return "same value element in ansVec[]";
}
sot.insert(v);
}
// array check
set<int>bset;
for (int i = 1; i <= m; ++i) {
bset.insert(b[i]);
}
int nxtPos = 1;
int bnum = 0;
for(int i = 0; i < n + m; ++i) {
if (nxtPos <= n && a[nxtPos] == ansVec[i]) {
++nxtPos;
bnum = 0;
}
else {
if(!bset.count(ansVec[i])) {
return "it is a wrong final array %d";
}
bset.erase(ansVec[i]);
if (++bnum > 1) {
return "can not insert continuous elements of array b";
}
}
}
return "AC";
}
}checker;
void printVec(string str, vector<LL>vec) {
if (str != "") printf("%s: ", str.c_str());//cout << str << ": ";
for(auto &it : vec) {
printf("%lld ", it);
}puts("");
}
void printVec(string str, vector<int>vec) {
if (str != "") printf("%s: ", str.c_str());//cout << str << ": ";
for(auto &it : vec) {
printf("%d ", it);
}puts("");
}
void HumanData()
{
srand(time(0));
freopen("Human Data.in", "w", stdout);
casenum = 8; printf("%d\n", casenum);
for (casei = 1; casei <= casenum; ++casei) {
set<int>sot;
n = 600; m = 601;
printf("%d %d\n", n, m);
int basic = 5000 + rand() % 1000;
int dif = rand() % 3 + 2;
a[1] = basic; sot.insert(a[1]);
a[2] = basic + 1; sot.insert(a[2]);
for(int i = 3; i <= n; ++i){
if (i & 1)a[i] = a[i - 2] - dif;
else a[i] = a[i - 2] + dif;
sot.insert(a[i]);
}
for(int i = 1; i <= n; ++i) {
printf("%d ", a[i]);
}puts("");
for(int i = 1; i <= m; ++i) {
do{
b[i] = rand() % (basic + dif * n) + 1;
}while(sot.count(b[i]));
sot.insert(b[i]);
printf("%d ", b[i]);
}puts("");
}
}
int main() {
// HumanData(); return 0;
// dataGenerator.rd_dataGenerator(); return 0;
freopen("Blood Pressure Game.in", "r", stdin); freopen("Blood Pressure Game.out", "w", stdout);
// freopen("Human Data.in", "r", stdin);
scanf("%d", &casenum); for(casei = 1; casei <= casenum; ++casei) {
printf("Case #%d:\n", casei);
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i) scanf("%d", &a[i]);
for (int i = 1; i <= m; ++i) scanf("%d", &b[i]);
LL basic = 0;
for (int i = 2; i <= n; ++i) {
basic += abs(a[i] - a[i - 1]);
}
// solution 1: fastNetworkFlow
bool NplusM_MCMF = true;
vector<LL> fastMCMFvec;
if (NplusM_MCMF) {
mcmf.NplusM_mapBuild();
fastMCMFvec = mcmf.fastMCMF(basic);
printVec("", fastMCMFvec); //fastMCMFvec
mcmf.NplusM_output();
FMS(mcmf.first, 0, mcmf.ED + 2);
// ans checker
string check_str = checker.check_it();
if(check_str != "AC") {
puts(check_str.c_str());
for (int i = 1; i <= n; ++i) {
printf("%d ", a[i]);
}puts("");
for (int i = 1; i <= m; ++i) {
printf("%d ", b[i]);
}puts("");
printf("SUM = %lld\n", mcmf.SUM);
while(true);
}
if(DEBUG) {
mcmf.printMap();
}
}
// solution 2: slowNetworkFlow
bool NmultM_MCMF = false;
vector<LL> slowMCMFvec;
if (NmultM_MCMF) {
mcmf.NmultM_mapBuild();
slowMCMFvec = mcmf.slowMCMF(basic);
printVec("", slowMCMFvec); // slowMCMFvec
mcmf.NmultM_output();
FMS(mcmf.first, 0, mcmf.ED + 2);
if (NplusM_MCMF && slowMCMFvec != fastMCMFvec) {
puts("Error: slowMCMFvec != fastMCMFvec");
while(true);
}
}
// solution 3: Brute Force (DFS)
bool bruteForce = false;
if (bruteForce) {
vector<LL> BFvec = bf.solve();
printVec("BFvec", BFvec);
bf.output();
if (NplusM_MCMF && BFvec != fastMCMFvec) {
puts("Error: BFvec != fastMCMFvec");
while(true);
}
}
// solution 4: Greedy
// LL ans_greedy = greedy.solve();
}
return 0;
}
/*
【Trick && Tsukkomi】
Input
1
4 5
21 3 48 39
16 66 9 64 36
Output
36 21 66 3 64 48 9 39 16
Input
1
4 4
10 50 3 6
1 9 23 5
Output
150
5 10 1 50 3 23 6 9
【题意】
把 m 个数任意插入到长度为 n 的数组中的缝隙或两侧,在每个位置最多只能插入一个数的条件下,使得相邻数之差的绝对值的和尽可能大。
【分析】
这道题,题目是将 m 个待插入数值,向 n + 1 个区间做插入。而这个插入其实也就是匹配。这种带权匹配问题,我们可以使用网络流(费用流)算法来解决。
因为我们希望最终的差值之和尽可能大,所以这个"费用"此处是"收益",我把它称呼为最大收益最大流好啦。
因为匹配的可能是 n * m ,这个图实际构成了"完全二分图",边数是m * (n + 1).
然而,面对1000的数据规模,O(nm) 的边数就已经巨大无比了,最终算法的复杂度将会难以吃得消。
要怎么办才好呢?我们可以结合这道题的特殊性,优化建图!
可以看到——除了首尾这两个特殊的插入位置外,其他所有的插入位置都可以用一个二元对[l, r]来表示。
如果插入的数值 v 比 l 小,其实收益只与 l 有关,是 (l - v) * 2
如果插入的数值 v 比 r 大,其实收益只与 r 有关,是 (v - r) * 2
否则,插入的数值在区间内,则该插入操作不会产生任何收益。
显然,我们发现,对于插入区间,笼统来说,是具有 l 越大优、或 r 越小越优的性质的。
其实也就是说,如果我们最终做了匹配 v < [l2, r2],那么不可能我们有一个闲置未匹配区间[l1, r1](l1 > l2)的,这样 v 匹配[l1, r1]一定更优。
同理, 如果我们最终做了匹配 [l2, r2] > v,那么不可能我们有一个闲置未匹配区间[l1, r1](r1 < r2)的,这样 v 匹配[l1, r1]一定更优
而对于两个区间,其替换后价值的收益其实是线性的。如 v < [l2, r2],由[l2, r2]调整为[l1, r1](l1 > l2)的时候,收益是(l1 - l2) * 2
发现了这些性质后,我们就可以优化建图啦——
(1) 设置源点编号为0,汇点编号为 m + n * 3 + 1
(2) 源点连接 m 个被插入点[1, m], 容量统一为1,收益统一为0 <此处m条边>
(3) 把所有非两侧的可插入区间,抽象为[m + 1, m + n)这 n - 1 个点,
按照左界从大到小(从优到差)排序,我们考虑每个区间都可能匹配(插入)了比它小的数值(v < l)
从区间 [l[i + 1], r[i + 1]] 向区间 [l[i], r[i]] 连一条容量极大(>=m即可),收益为(l[i] - l[i + 1]) * 2的边 <此处n - 2条边>
(4) 把所有非两侧的可插入区间,抽象为[m + n + 1, m + n + n)这 n - 1 个点,
按照右界从小到大(从优到差)排序,我们考虑每个区间能可能匹配(插入)了比它大的数值(v > r)
从区间 [l[i + 1], r[i + 1]] 向区间 [l[i], r[i]] 连一条容量极大(>=m即可),收益为(r[i + 1] - r[i]) * 2的边 <此处n - 2条边>
(5) 然而,一个区间最多只能匹配一次,即不可能其既作为较大的区间被插入了值,同时由作为最小的区间被插入了值。
因此,我们再设置 [m + n + n + 0, m + n + n + n] 这 n + 1 个点,这些点向汇点 ED 连一条容量为1,收益为0的边 <此处n + 1条边>
同时,对于i ∈ [1, n), m + i 与 m + n + i 同时向 m + n + n + i 连一条容量为1(>=1即可),收益为0的边 <此处(n - 1) * 2条边>
于是,我们控制使得每个区间被最多匹配一次,同时一个区间不可能同时作为较大区间和较小区间同时被匹配插入。
(6) 注意到,可以被插入匹配的位置其实有 n + 1 个,而对于首区间和尾区间,编号实际为 m + n + n + 0 和 m + n + n + n,
我们直接从 m 个插入点向这 2 个特殊插入位置连一条容量为1(>=1即可),收益为0的边 <此处m * 2条边>
(7) 不要忘记了,"向下匹配的左界区间"和"向上匹配的右界区间",虽然它们都被连成了链,且连入了唯一编号的区间,但却没有流量流入。
对于 m 个插入值 v ,向比其大的第一个(如果存在) "下匹配的左界区间[l, r]", 连一条容量为1(>=1即可),收益为(l - v) * 2的边
同理, 向比其小的第一个(如果存在) "上匹配的右界区间[l, r]", 连一条容量为1(>=1即可),收益为(v - r) * 2的边
<此处最多m * 2条边>
这个图最终形成啦。
层次包括六层{源点0}、{插入层[1, m]}、{下匹配左界区间层[m + 1, m + n)}、{上匹配右界区间层[m + n + 1, m + n + n)}、{真实区间层[m + n + n + 0, m + n + n + n]}、{汇点m + n * 3 + 1}
点数总共2 + m + (n - 1) + (n - 1) + (n + 1)共计m + n * 3 + 1个,即点数为4n级别
同时,边数可以由(1)~(7)求和可得,为10n级别
因而,算法的复杂度为O(点数为4n边数为10n的费用流复杂度) :p
【数据】
6
2 3
5 11
10 3 1
4 1
1 2 3 4
5
4 2
1 2 3 4
5 6
4 5
1 2 3 4
5 6 7 8 9
4 4
10 50 3 6
1 9 23 5
4 2
10 50 3 6
9 23
*/