题目链接:http://www.lydsy.com/JudgeOnline/problem.php?id=4882
Description
小Q最近沉迷底层优化,他终于将他的MST模板优化到了很可怕的地步,所以他出了这道题来试验模板的速度。给定
一个n个点的带权无向连通图,点的编号从1到n,请求出这个图的最小生成树。为了避免输入数据过于庞大,数据
进行了压缩,解压方式可用下面这段代码解释:
void encode(int seed, int n){
int x = seed;
for(int i = 2; i <= n; i++){
x = x * 907 % 2333333;
int T = x;
for(int j = max(1, i - 5); j <= i - 1; j++){
x = x * 907 % 2333333;
int w = T ^ x;
add_edge(i, j, w);//在点i与点j之间添加一条边权为w的边
}
}
}
Input
第一行包含一个正整数T(1<=T<=50),表示测试数据的组数。
每组数据只包含一行两个正整数n,seed(1<=n<=10^7,1<=seed<=2333332),含义如题面所述。
Output
对于每组数据,输出一行一个整数,即最小生成树的树边的权值之和。
Sample Input
3
3010 2016
3 14159
31415 926
3010 2016
3 14159
31415 926
Sample Output
750887251
3382896
9210525875
3382896
9210525875
观察模数 2333333 可以发现权值的循环节不会很大,大约是 50 的级别。
考虑暴力状压 DP,设f[i][mask] 表示考虑前 i 个点,最后 5 个点的连通性为mask 时的最小代价,那么 f[i][] f [i 1][] 的循环节也必然不会很大。暴力找到循环节,跳过循环节,最后零碎部分再暴力 DP 即可。
//找循环节然后暴力状压DP
#include<stdio.h>
#include<algorithm>
#include<math.h>
#include<string.h>
#include<string>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<time.h>
#include<assert.h>
#include<iostream>
using namespace std;
typedef long long LL;
typedef pair<int,int>pi;
const int Ms=3125;
const LL Inf=1LL<<60;
const int LIM=100;
int n,seed;
LL in[LIM+1];
LL f[LIM+1][Ms];
LL dp[2][Ms];
void init(int cs,LL val){
for(int i=0;i<Ms;i++)dp[cs][i]=val;
}
int g[Ms][(1<<5)+1];
void decode(int mask,int *a){
for(int i=0;i<5;i++){
a[i]=mask%5;
mask/=5;
}
}
void zuixiao(int *a){
int dui[10];
memset(dui,-1,sizeof dui);
int cur=0;
for(int i=0;i<5;i++){
int x=a[i];
if(dui[x]<0){
dui[x]=cur++;
}
a[i]=dui[x];
}
}
int encode(int *a){
int res=0,tmp=1;
for(int i=0;i<5;i++,tmp=tmp*5){
res+=a[i]*tmp;
}
return res;
}
bool testt(int *a){
int b[10];
for(int i=0;i<5;i++)b[i]=a[i];
zuixiao(b);
for(int i=0;i<5;i++)if(b[i]!=a[i])return 0;
return 1;
}
void pre(){
int a[10],b[10];
for(int mask=0;mask<Ms;mask++){
decode(mask,a);
if(!testt(a)){
for(int j=0;j<1<<5;j++)g[mask][j]=-1;
continue;
}
for(int j=0;j<1<<5;j++){
for(int i=0;i<5;i++)b[i]=a[i];
b[5]=6;
for(int k=0;k<5;k++){
if(j>>k&1){
for(int t=0;t<5;t++){
if(t==k)continue;
if(b[t]==b[k])b[t]=b[5];
}
b[k]=b[5];
}
}
bool flag=0;
for(int k=1;k<=5;k++)if(b[0]==b[k]){flag=1;break;}
if(!flag)g[mask][j]=-1;
else{
for(int k=0;k<5;k++)b[k]=b[k+1];
zuixiao(b);
//for(int k=0;k<5;k++)printf("%d ",b[k]);puts("");
g[mask][j]=encode(b);
}
}
}
}
void upd(LL &x,LL y){if(x>y)x=y;}
int main(){
pre();
int _;scanf("%d",&_);
while(_--){
scanf("%d%d",&n,&seed);
int cs=0;
init(cs,Inf);
dp[cs][0]=0;
for(int i=2;i<=n&&i<=LIM;i++){
init(cs^1,Inf);
seed=seed*907%2333333;
LL bq[10];
int T=seed;
in[i]=T;
for(int j=0;j<5;j++){
if(i-(5-j)<1)bq[j]=Inf;
else {
seed=seed*907%2333333;
bq[j]=seed^T;
}
}
for(int mask=0;mask<Ms;mask++){
LL w=dp[cs][mask];
if(w==Inf)continue;
for(int j=0;j<1<<5;j++){
if(g[mask][j]<0)continue;
bool flag=1;
for(int k=0;k<5;k++)if((j>>k&1)&&(bq[k]==Inf)){
flag=0;continue;
}
if(!flag)continue;
LL nw=w;
for(int k=0;k<5;k++){
if(j>>k&1)nw+=bq[k];
}
upd(dp[cs^1][g[mask][j]],nw);
}
}
cs^=1;
for(int j=0;j<Ms;j++)f[i][j]=dp[cs][j];
}
if(n<=LIM)printf("%lld\n",dp[cs][0]);
else{
LL pace,pacew;
for(int i=LIM-1;;i--)if(in[i]==in[LIM]){
pace=LIM-i;
break;
}
int st=(n-LIM)/pace*pace+LIM;
for(int i=0;i<Ms;i++){
dp[cs][i]=f[LIM][i]+1LL*(n-LIM)/pace*(f[LIM][i]-f[LIM-pace][i]);
}
for(int i=st+1;i<=n;i++){
init(cs^1,Inf);
seed=seed*907%2333333;
LL bq[10];
int T=seed;
for(int j=0;j<5;j++){
if(i-(5-j)<1)bq[j]=Inf;
else {
seed=seed*907%2333333;
bq[j]=seed^T;
}
}
for(int mask=0;mask<Ms;mask++){
LL w=dp[cs][mask];
if(w==Inf)continue;
for(int j=0;j<1<<5;j++){
if(g[mask][j]<0)continue;
bool flag=1;
for(int k=0;k<5;k++)if((j>>k&1)&&(bq[k]==Inf)){
flag=0;continue;
}
if(!flag)continue;
LL nw=w;
for(int k=0;k<5;k++){
if(j>>k&1)nw+=bq[k];
}
upd(dp[cs^1][g[mask][j]],nw);
}
}
cs^=1;
}
printf("%lld\n",dp[cs][0]);
}
}
return 0;
}