Problem
给定序列A,序列中的每一项Ai有删除代价Bi和附加属性Ci。请删除若干项,使得A的最长上升子序列长度减少至少1,且付出的代价之和最小,并输出方案。 如果有多种方案,请输出将删去项的附加属性排序之后,字典序最小的一种。
Solution
- 考虑建立流量网络:对序列中的每一项进行拆点连边,i to i+n容量为bi,dp(i)表示以序列第i项结尾的最长上升子序列长度,沿最长上升子序列连边,若j<i且a[i]>a[j]且dp(i)=dp(j)+1,j+n to i连容量为inf的边,源点向所有dp值为1的点连容量为inf的边,所有dp值为最长上升子序列长度的点向汇点连容量为inf的边。
- 考虑最小割:该模型中限定了最小割中切断的边∈{(i,i+n),i<=n},最小割为源点至汇点不存在可行流所需付出的最小代价,即删点后使得不存在原最长上升子序列长度的子序列所需最小代价。
- 字典序最小方案:按c值排序,依次考虑该边(u,v)能否成为满足当前条件(c值比当前边小的最小割可行边全部割)下最小割的可行边,具体方法是对u在残量网络中bfs,看是否存在到v的可行流,若不存在则(u,v)为最小割可行边,加进答案中,强制割(u,v)容量置0,重新跑dinic得到残量网络。
- 最小割可行边:若(u,v)为最小割可行边,则残量网络中(u,v)满流且不存在u到v的可行流。若(u,v)未满流则对于s到u到v到t的可行流所割的边不为(u,v);若存在u到v的可行流则割掉(u,v)之后仍存在s到u到v到t的可行流。
- 最小割必须边:若(u,v)为最小割必须边,则(u,v)为最小割可行边且残量网络中同时存在s到u的可行流和v到t的可行流。必须性:若不割(u,v),必不可能以割(u,v)的代价割其它边使得s到u到v到t没有可行流。
Code
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int maxn = 1505,inf = 1e9;
int n,cnt = 1,last[maxn],ceng[maxn],dt,dp[maxn],S,T,cur[maxn],mx,pos[maxn],cnt0[maxn];
struct edge{
int v,w,next;
}e[maxn*maxn];
inline void add(int u,int v,int w)
{
e[++cnt] = {v,w,last[u]};
last[u] = cnt;
e[++cnt] = {u,0,last[v]};
last[v] = cnt;
}
struct ele{
int num,a,b,c;
}el[maxn];
bool cmp(const ele &a,const ele &b){return a.c<b.c;}
bool bfs(int s,int t)
{
std::queue<int> q;
memset(ceng,0,sizeof(ceng));
ceng[s]=1;
q.push(s);
while(!q.empty())
{
int u=q.front();q.pop();
for(int i=last[u];i;i=e[i].next)
{
int v=e[i].v,w=e[i].w;
if(ceng[v]==0&&w>0)
{
ceng[v]=ceng[u]+1;
q.push(v);
if(v==t) return 1;
}
}
}
return ceng[t];
}
int dfs(int u,int dis,int t)
{
if(u==t||!dis)
return dis;
int ff = 0;
for(int i=cur[u];i;i=e[i].next)
{
cur[u]=e[i].next;
int v=e[i].v,w=e[i].w;
if(ceng[v]==ceng[u]+1&&w>0)
{
int di=dfs(v,min(dis,w),t);
if(di<=0) continue;
e[i].w-=di;
e[i^1].w+=di;
ff+=di;
dis-=di;
if(!dis) break;
}
}
return ff;
}
long long dinic()
{
long long ans=0;
while(bfs(S,T))
{
for(int i=S;i<=T;i++)
cur[i]=last[i];
ans+=dfs(S,inf,T);
}
return ans;
}
void init()
{
mx = S = 0;T = n*2+1;
cnt = 1;
for(int i=1;i<=n;i++)
{
el[i].num = i;
dp[i] = 1;
}
memset(last,0,sizeof(last));
}
int main()
{
cin>>dt;
while(dt--)
{
scanf("%d",&n);
init();
for(int i=1;i<=n;i++) scanf("%d",&el[i].a);
for(int i=1;i<=n;i++) scanf("%d",&el[i].b);
for(int i=1;i<=n;i++) scanf("%d",&el[i].c);
for(int i=1;i<=n;i++)
{
for(int j=1;j<i;j++)
if(el[i].a>el[j].a)
dp[i] = max(dp[i],dp[j]+1);
mx = max(mx,dp[i]);
}
for(int i=1;i<=n;i++)
{
add(i,i+n,el[i].b);
cnt0[i] = cnt-1;
}
for(int i=1;i<=n;i++)
{
if(dp[i]==1)
add(0,i,inf);
if(dp[i]==mx)
add(i+n,T,inf);
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<i;j++)
{
if(dp[i]==dp[j]+1&&el[i].a>el[j].a)
{
add(j+n,i,inf);
// cout<<j<<' '<<"to"<<' '<<i<<endl;
}
}
}
long long ans = dinic(),numpos = 0;
std::sort(el+1,el+1+n,cmp);
for(int i=1;i<=n;i++)
{
if(!bfs(el[i].num,el[i].num+n))
{
e[cnt0[el[i].num]].w = e[cnt0[el[i].num]^1].w = 0;
for(int j=2;j<=cnt;j+=2)
{
e[j].w+=e[j^1].w;
e[j^1].w = 0;
}
while(bfs(S,T))
{
for(int j=S;j<=T;j++)
cur[j] = last[j];
dfs(S,inf,T) ;
}
pos[++numpos] = el[i].num;
// cout<<i<<' '<<el[i].a<<' '<<el[i].b<<' '<<el[i].c<<endl;
}
}
cout<<ans;
std::sort(pos+1,pos+1+numpos);
cout<<' '<<numpos<<endl<<pos[1];
for(int i=2;i<=numpos;i++)
cout<<' '<<pos[i];
cout<<endl;
}
return 0;
}