傳送門
文章目錄
題意：
給你一個長度為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)
輸出方案的時候記錄一下組數就好啦，沒有要求組數最少隨便找一組即可。
#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
;
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()
{
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;
}
總結
以上是生活随笔為你收集整理的Codeforces Round #598 (Div. 3) E. Yet Another Division Into Teams dp + 输出方案的全部內容，希望文章能夠幫你解決所遇到的問題。
如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。
