天天看點

牛客練習賽44----C-小y的質數

首先發出題目連結:

連結:https://ac.nowcoder.com/acm/contest/634/C

來源:牛客網

涉及:容斥定理,gcd及擴充gcd

題目如下:

牛客練習賽44----C-小y的質數
牛客練習賽44----C-小y的質數
牛客練習賽44----C-小y的質數

一般來說看到這種gcd數學題看到資料範圍折磨大就知道肯定有技巧!然而這個題真的有技巧。

要知道,題目意思是讓我們在區間[L+K,R-K]裡面找出所有y的個數,使得y+k和y-k互質

這個問題看似隻能用暴力,但是其實可以把這個問題等價代換一下,變成:

讓我們在區間[L,R-2K]裡面找到所有y的個數,使得y和y+2k互質。

即 g c d ( y , y + 2 ∗ k ) = 1 gcd(y,y+2*k)=1 gcd(y,y+2∗k)=1

而 g c d ( y , y + 2 ∗ k ) = 1 gcd(y,y+2*k)=1 gcd(y,y+2∗k)=1又可以變成 g c d ( y , 2 ∗ k ) = 1 gcd(y,2*k)=1 gcd(y,2∗k)=1

是以最後問題變成了

讓我們在區間[L,R-2K]裡面找到所有y的個數,使得y和2k互質,是以對于y來說,y的取值範圍就是[L,R-2K],可以在開始就把R-2K賦給R。

特判:如果R-2*K<L,則答案為0;

要是y和2k互質,則y和2k的最大公約數隻能為1

根據可知,所有的正整數都可以化為n個素數的乘積,是以們可以把2k分解質因子,得到2k的所有質因子,然後求出[L,R-2K]所有的質因子的倍數的個數x,然後用區間的長度減去x,得到的就是區間内和2k互質的數的個數

而求出k的所有質因子的個數,可以用容斥定理來求得

容斥定理:

( A 1 ∪ A 2 ∪ A 3 ∪ . . . ∪ A n ) = ∑ 0 &lt; i ≤ n A i − ∑ 0 &lt; i &lt; j ≤ n ( A i ∩ A j ) + ∑ 0 &lt; i &lt; j &lt; k ≤ n ( A i ∩ A j ∩ A j ) + . . . + ( − 1 ) n + 1 ( A 1 ∩ A 2 ∩ A 3 ∩ . . . ∩ A n ) (A_{1} \cup A_{2} \cup A_{3} \cup ... \cup A_{n})=\sum_{0&lt;i\le n} A_{i}-\sum_{0&lt;i&lt;j\le n} (A_{i} \cap A_{j})+\sum_{0&lt;i&lt;j&lt;k\le n}(A_{i}\cap A_{j}\cap A_{j})+...+(-1)^{n+1}(A_{1} \cap A_{2}\cap A_{3}\cap ... \cap A_{n}) (A1​∪A2​∪A3​∪...∪An​)=0<i≤n∑​Ai​−0<i<j≤n∑​(Ai​∩Aj​)+0<i<j<k≤n∑​(Ai​∩Aj​∩Aj​)+...+(−1)n+1(A1​∩A2​∩A3​∩...∩An​)

将2k分解質因子

由于k的資料範圍小于1013,分解出來的質數頂多也就12個左右,可以用暴力求得k的所有質數

先将2k的值賦給k,然後i從2開始往上找,是i的因數那就隻可能i是質數,然後while循環k除以i,将i放到數組裡備用,最後for循環結束之後,如果k大于1,此時的k也是原來2k的一個質因子,最後divisor數組裡就存的是原來2k的所有質因子。

void resolve(){
	int i;
	k=2*k;
	for(i=2;i*i<=k;i++){
		if(k%i==0)	divisor[tot++]=i;
		while(k%i==0)	k/=i;
	}
	if(k!=1)	divisor[tot]=k;
	return;
}
           

容斥定理

容斥定理偶數個集合交集就是容,奇數個集合交集就是斥

集合交集就是多個質因子相乘後的數的倍數的個數(質因子互質,最小公倍數就是相乘)

void dfs(ll p,ll x,ll mu,bool ok){
	if(ok && x)	sum+=mu*(r/x-(l-1>=0?l-1:0)/x);
	if(p==tot+1)	return;
	dfs(p+1,x,mu,false);
	dfs(p+1,x*divisor[p],-mu,true);
	return;
}
           

容斥定理可以用遞歸來實作

當最小公倍數是偶數個質因子相乘即是斥,反之則容

每個質因子可以考慮乘以目前的公倍數得到新的公倍數,也可以不乘。

dfs(ll p,ll x,ll mu,bool ok)

其中p是考慮第p-1個質因子是否與目前公倍數相乘;

x是目前的公倍數(開始遞歸的時候x要取1,ok要取false);

mu是判斷容斥;

ok是防止重複計算(當考慮到某個質因子不與目前公倍數相乘,則下次遞歸x未發生變化,就不用再次計算此公倍數的倍數在區間内的個數)

計算目前公倍數的倍數在區間内的個數是 s u m + = m u ∗ ( r / x − ( l − 1 ) / x ) sum+=mu*(r/x-(l-1)/x) sum+=mu∗(r/x−(l−1)/x)(L是原來的l減去了2k)

(r/x是1到r内x的倍數的個數,(l-1)是1到(l-1)内公倍數的個數,兩個相減就是區間内x倍數的個數)

ps:這個式子隻能當L和R大于1時候成立,不然個數就少算了一個(不行你可以去試一試)

還要注意一點,由于L的資料範圍是大于等于0,而0與任何數的最小公因數是這個數本身,由于1不是質數,是以如果輸入的L是等于0的話,sum少算了一個數0,最終sum要加上1;

是以,就求個數的式子應該是

m u ∗ ( r / x − ( l − 1 &gt; = 0 ? l − 1 : 0 ) / x ) mu*(r/x-(l-1&gt;=0?l-1:0)/x) mu∗(r/x−(l−1>=0?l−1:0)/x),當L為0時,L-1要仍然為0;

遞歸結束後,答案就是區間長度減去sum(sum是區間内與2k不互質數的個數)

r − l + 1 − ( s u m + ( l = = 0 ? 1 : 0 ) ) r-l+1-(sum+(l==0?1:0)) r−l+1−(sum+(l==0?1:0))(當L為0時,最終sum要加上1)

日常舉例子

就舉0 20 6這個例子

首先将6分解質因子

分解出2和3

l=0,r=r-2*k=8

開始dfs(1,1,-1,false)//dfs(ll p,ll x,ll mu,bool ok)

1.在dfs(1,1,-1,false)中,ok等于false,跳過中間步驟

下一下語句是dfs(2,1,-1,false)和dfs(2,2,1,true)

1.1在dfs(2,1,-1,false)中,ok等于false,跳過中間步驟

下一條語句是dfs(3,1,-1,false)和dfs(3,3,-1,true)

1.1.1在dfs(3,1,-1,false)中,ok等于false,跳過中間步驟

下一條語句是if(p==tot+1) return;

終止遞歸(tot是質因子的數量,tot=2)

1.1.2在dfs(3,3,-1,true)中,ok等于true

此時sum=0,計算sum+=mu*(r/x-(l-1>=0?l-1:0)/x);

(求8以内3的倍數個數,算出結果為8/3=2,此時sum=2)

下一條語句是if(p==tot+1) return;

終止遞歸

1.2在dfs(2,2,1,true)中,ok等于true

此時sum=2,計算sum+=mu*(r/x-(l-1>=0?l-1:0)/x);

(求8以内2的倍數個數,算出結果為8/2=4,此時sum=6)

下一條語句是dfs(3,2,1,false)和dfs(3,6,-1,true)

1.2.1在dfs(3,2,1,false)中,ok等于false,跳過中間步驟

下一條語句是if(p==tot+1) return;

終止遞歸(tot是質因子的數量,tot=2)

1.2.2在dfs(3,6,-1,true)中,ok等于true

此時sum=6,計算sum+=mu*(r/x-(l-1>=0?l-1:0)/x);

(求8以内6的倍數個數,算出結果為8/6=1,此時sum=5)

下一條語句是if(p==tot+1) return;

終止遞歸(tot是質因子的數量,tot=2)

最終sum=5,由于L=0,sum=sum+1=6;

答案為r-l+1-sum=8-0+1-6=3

代碼如下:

#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
ll l,r,k,tot=1;//l和r以及k是題目資料,tot是k的質因子的種類個數
ll divisor[25]={0};//這個數組存k的質因子
ll sum=0;
void resolve(){//k分解質因子函數
	int i;
	for(i=2;i*i<=k;i++){
		if(k%i==0)	divisor[tot++]=i;
		while(k%i==0)	k/=i;//防止重複算質因子
	}
	if(k!=1)	divisor[tot]=k;//最後的k如果不為0,也仍然是一個質因子
	return;
}

void dfs(ll p,ll x,ll mu,bool ok){//容斥定理
	if(ok && x)	sum+=mu*(r/x-(l-1>=0?l-1:0)/x);
	if(p==tot+1)	return;//所有質因子全部被考慮到就傳回
	dfs(p+1,x,mu,false);//不選擇divisor[p]乘目前公倍數x,則x不變,下次遞歸跳過計算步驟
	dfs(p+1,x*divisor[p],-mu,true);//選擇divisor[p]乘目前公倍數x,則x變化,下次遞歸以新的x再次計算x的倍數在區間的個數
	return;
}

int main(){
	cin>>l>>r>>k;
	r-=2*k;
	if(l>r)	return !(cout<<"0");//特判
	k=2*k;
	resolve();
	dfs(1,1,-1,false);//從第一個質因子開始考慮,x此時為1,令mu=-1,由于x=1時不知質因子公倍數,是以不進行計算
	cout<<r-l+1-(sum+(l==0?1:0));//如果l為0,則sum要加上1(要考慮特殊數字0)
	return 0;
}
           

繼續閱讀