传送门
文章目录
- 题意:
- 思路:
题意:
给你一个长度为nnn的序列aaa,你需要将其分成若干组,每组的价值为max(ai)−min(ai)\max(a_i)-min(a_i)max(ai)−min(ai),求如何分组才能使得代价最小,输出最小代价以及分组方案。
3≤n≤2e5,1≤ai≤1e93\le n\le2e5,1\le a_i\le1e93≤n≤2e5,1≤ai≤1e9
思路:
首先可以将这个数组排个序,不过要在排序之前记录一下编号。
排序之后就可以大力dpdpdp了,定义f[i]f[i]f[i]表示到了第iii个数的最小代价,那么肯定是从[1,i−3][1,i-3][1,i−3]转移过来,但是这样转移是n2n^2n2的,需要优化一下。
通过多写几组可以发现,我们如果分出来了个长度为666的组,那么把它分成俩长度为333的组答案一定不会变劣。所以我们只需要从[i−6,i−3][i-6,i-3][i−6,i−3]转移过来,那么转移方程就是f[i+j]=min(f[i+j],f[i−1]+a[i+j].val−a[i].val)f[i+j]=min(f[i+j],f[i-1]+a[i+j].val-a[i].val)f[i+j]=min(f[i+j],f[i−1]+a[i+j].val−a[i].val)
输出方案的时候记录一下组数就好啦,没有要求组数最少随便找一组即可。
// Problem: E. Yet Another Division Into Teams
// Contest: Codeforces - Codeforces Round #598 (Div. 3)
// URL: https://codeforces.com/contest/1256/problem/E
// Memory Limit: 256 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4.1,sse4.2,avx,avx2,popcnt,tune=native")
//#pragma GCC optimize(2)
#include<cstdio>
#include<iostream>
#include<string>
#include<cstring>
#include<map>
#include<cmath>
#include<cctype>
#include<vector>
#include<set>
#include<queue>
#include<algorithm>
#include<sstream>
#include<ctime>
#include<cstdlib>
#include<random>
#include<cassert>
#define X first
#define Y second
#define L (u<<1)
#define R (u<<1|1)
#define pb push_back
#define mk make_pair
#define Mid ((tr[u].l+tr[u].r)>>1)
#define Len(u) (tr[u].r-tr[u].l+1)
#define random(a,b) ((a)+rand()%((b)-(a)+1))
#define db puts("---")
using namespace std;//void rd_cre() { freopen("d://dp//data.txt","w",stdout); srand(time(NULL)); }
//void rd_ac() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//AC.txt","w",stdout); }
//void rd_wa() { freopen("d://dp//data.txt","r",stdin); freopen("d://dp//WA.txt","w",stdout); }typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;const int N=1000010,mod=1e9+7,INF=0x3f3f3f3f;
const double eps=1e-6;int n;
LL f[N];
int ans[N],tot;
struct Node {int val,id;bool operator < (const Node &W) const {return val<W.val;}
}a[N];int main()
{
// ios::sync_with_stdio(false);
// cin.tie(0);scanf("%d",&n);for(int i=1;i<=n;i++) scanf("%d",&a[i].val),a[i].id=i;sort(a+1,a+1+n);memset(f,INF,sizeof(f));f[0]=0;for(int i=1;i<=n;i++) {for(int j=2;j<=5;j++) f[i+j]=min(f[i+j],f[i-1]+a[i+j].val-a[i].val);}for(int i=n;i>=1;) {for(int j=3;j<=5;j++) if(i-j>=0&&f[i]==f[i-j]+a[i].val-a[i-j+1].val) {tot++;for(int h=i-j+1;h<=i;h++) ans[a[h].id]=tot;i-=j;break;}}cout<<f[n]<<' '<<tot<<endl;for(int i=1;i<=n;i++) printf("%d ",ans[i]); puts("");return 0;
}
/**/