BZOJ4882: [Lydsy2017年5月月赛]卡常生成树

题目链接: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

Sample Output

750887251
3382896
9210525875

观察模数 2333333 可以发现权值的循环节不会很大,大约是 50 的级别。

 

考虑暴力状压 DP,设f[i][mask表示考虑前 i 个点,最后个点的连通性为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;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值