BZOJ2118: 墨墨的等式 思维建图

问a1x1+a2x2+a3x3+……+anxn∈[Bmin,Bmax]中的非负整数解有多少组

由于是非负所以不能简单求gcd,可以考虑任取其中某一项ax,

以它的模域0-ax-1建图,单向通过a数组转移,这样跑出从模0到模任意数的最短路

最短是为了保证在BminBmax里能塞进最多的ax

显然ax取最小可以让复杂度最好,实测取max是min耗时2倍

正确性就是如果m可以取到,那么k*ax+m都可以取到,只要数区间内有多少ax倍数就行,这个用一下前缀和思想就行

至于为什么能任取ax,因为当ax变大,模域也变大,转移路径也会变多,不会因为区间内ax倍数变少而影响正确性

//#include<bits/stdc++.h>  
//#pragma comment(linker, "/STACK:1024000000,1024000000")   
#include<stdio.h>  
#include<algorithm>  
#include<queue>  
#include<string.h>  
#include<iostream>  
#include<math.h>  
#include<set>  
#include<map>  
#include<vector>  
#include<iomanip>  
using namespace std;  
  
const double pi=acos(-1.0);  
#define ll long long  
#define pb push_back

#define sqr(a) ((a)*(a))
#define dis(a,b) sqrt(sqr(a.x-b.x)+sqr(a.y-b.y))

const double eps=1e-10;
const int maxn=6e6+56;
const int inf=0x3f3f3f3f;
const ll mod=6;

ll d[maxn];
int arr[maxn];
bool vis[maxn];

struct NODE{
	int d,u;
	bool operator < (const NODE &rhs)const{
		return d>rhs.d;
	}
};
struct EDGE{
	int u,v,w,nxt;
}G[maxn];int tot,head[maxn];
void addedge(int u,int v,int w){
	G[++tot]=(EDGE){u,v,w,head[u]};
	head[u]=tot;
}
void dij(int n){
	priority_queue<NODE>Q;
	Q.push((NODE){0,0});
	for(int i=1;i<=n;i++)d[i]=9e18;
	while(!Q.empty()){
		NODE x=Q.top();Q.pop();
		if(vis[x.u])continue;vis[x.u]=1;
		for(int i=head[x.u];~i;i=G[i].nxt){
			if(d[G[i].v]>d[x.u]+G[i].w){
				d[G[i].v]=d[x.u]+G[i].w;
				Q.push((NODE){d[G[i].v],G[i].v});
			}
		}
	}
}
int main(){
	memset(head,-1,sizeof head);
	int n;scanf("%d",&n);ll l,r;
	scanf("%lld%lld",&l,&r);l--;
	int mn=inf;
	for(int i=1;i<=n;i++){
		scanf("%d",&arr[i]);
		mn=min(mn,arr[i]);
	}

	for(int i=0;i<mn;i++){
		for(int j=1;j<=n;j++){
			if(arr[j]%mn!=0){
				addedge(i,(i+arr[j])%mn,arr[j]);
			}
		}
	}

	dij(mn-1);ll ans=0;		//计算每个模数互相转移的最少步数
	for(int i=0;i<mn;i++){
		if(d[i]<=l)ans-=(l-d[i])/mn+1;
		if(d[i]<=r)ans+=(r-d[i])/mn+1;
	}
	printf("%lld\n",ans);
}
学过平衡树算是明白最短路那个vis的意义了,虽说实测这题数据加了vis反而慢了。。
posted @ 2017-10-01 15:08  Drenight  阅读(205)  评论(0编辑  收藏  举报