Solution -「多校联训」朝鲜时蔬

\(\mathcal{Description}\)

  Link.

  破案了，朝鲜时蔬 = 超现实树！（指写得像那什么一样的题面。

  对于整数集 \(X\)，定义其 好子集 为满足 \(Y\subseteq X\land\left(\sum_{y\in Y}y\right)\mid\left(\sum_{x\in X}x\right)\) 的任意 \(Y\)。求 \(S_n=[1,n]\cap\mathbb N\) 的所有 \(m\) 阶子集中，包含 \(k\) 阶 好子集 数量最多的子集数。答案模 \((10^9+7)\)。

  \(k\le m\le n\le10^{12}\)，\(m\le4\)。

\(\mathcal{Solution}\)

\[\mathfrak{\text{Defining }\LaTeX\text{ macros...}}\newcommand{\vct}[1]{\boldsymbol{#1}}\newcommand{\stir}[2]{\genfrac{\{}{\}}{0pt}{}{#1}{#2}}\newcommand{\opn}[1]{\operatorname{#1}}\newcommand{\lcm}[0]{\opn{lcm}}\newcommand{\sg}[0]{\opn{sg}}\newcommand{\dist}[0]{\opn{dist}}\newcommand{\lca}[0]{\opn{lca}}\newcommand{\floor}[2]{\left\lfloor\frac{#1}{#2}\right\rfloor}\newcommand{\ceil}[2]{\left\lceil\frac{#1}{#2}\right\rceil}
\]

  令 \(f_{m,k}(n)\) 表示一组 \((m,k,n)\) 的答案，讨论之。

• \(m=k=1.\)

  显然 \(f_{m,k}(n)=n\)。

• \(m=2,k=1.\)

  设子集为 \(\{a,b\}\)，\(a<b\)，则应有 \(a\mid b\)。所以 \(f_{m,k}(n)=\sum_{i=1}^n\left(\floor{n}{i}-1\right)\)。

• \(m=k=2.\)

  显然 \(f_{m,k}(n)=\binom{n}{2}\)。

• \(m=3,k=1.\)

  必然有 \(\{k,2k,3k\}\) 为最优集合，即 \(f_{m,k}(n)=\floor{n}{3}\)。

  证明

  - 考虑集合 $\{x,y,z\}$，$x
  

• \(m=3,k=2.\)

  设子集 \(\{a,b,c\}\)，\(a<b<c\)，则应有 \((a+b)\mid c\)。枚举 \(a+b=i\)，得到

  \[f_{m,k}(n)=\sum_{i=1}^n\floor{i-1}{2}\floor{n}{i}.
  \]

• \(m=k=3.\)

  显然 \(f_{m,k}(n)=\binom{n}{3}\)。

• \(m=4,k=1.\)

  这个结论不太好正向推出啊。总之，当 \(n\ge6\)，最优集合 \(\{a,b,c,d\}\) 的形式必为以下一种：

  \[\{k,2k,3k,4k\},\\\{k,2k,6k,9k\},\\\{k,3k,8k,12k\},\\\{k,4k,5k,10k\},\\\{k,6k,14k,21k\},\\\{2k,3k,10k,15k\}.
  \]

  加之边界讨论，最终有

  \[f_{m,k}=\begin{cases}1,&n\le5\\\floor{n}{6}+\floor{n}{9}+\floor{n}{10}+\floor{n}{12}+\floor{n}{15}+\floor{n}{21},&n>5\end{cases}.
  \]
  证明

  对于最优的 $\{a,b,c,d\}$，$a2$：
  $$
  \begin{aligned}\frac{1}{x}+\frac{1}{y}+\frac{1}{z}+\frac{1}{w}&\le\frac{1}{3}+\frac{1}{4}+\frac{1}{5}+\frac{1}{6}\\&=0.95\\&\le1.\end{aligned}
  $$
  又显然 $x>1$，故 $x=2$。类似可证 $y\in\{3,4,5\}$。分别讨论即可解出 $z,w$。 $\square$
  

• \(m=4,k=2.\)

  \(n<11\) 可以打表，\(n\ge11\) 我们喜闻乐见地得到神奇结论：

  \[f_{m,k}(n)=\floor{n}{11}+\floor{n}{29},~~~~n\ge11.
  \]
  证明

  构造最优的 $\{a,b,c,d\}$，应有
  $$
  \begin{cases}
  (a+b)\mid(a+b+c+d)\Rightarrow (a+b)\mid(c+d),\\
  (a+c)\mid(a+b+c+d)\Rightarrow (a+c)\mid(b+d),\\
  \left.\begin{array}{}
  (a+d)\mid(a+b+c+d)\\
  (b+c)\mid(a+b+c+d)
  \end{array}\right\}\Rightarrow a+b=c+d.
  \end{cases}
  $$
  设 $k(a+c)=b+d=a+2b-c<3(a+c)$，知 $k=2$。类似讨论可证。 $\square$
  

• \(m=4,k=3.\)

  这个总算能阳间推导了 qwq。

  对于 \(\{a,b,c,d\}\)，必然 \((a+b+c)\mid d\)。枚举 \(a+b+c=i\)，记 \(f(i)\) 表示把 \(i\) 拆成 \(a,b,c\) 的方案数，不难发现

  \[f(i)=\binom{i-1}{2}-3\left(\ceil{n}{2}-1\right)+2[3\mid n].
  \]

  答案则为

  \[f_{m,k}(n)=\sum_{i=1}^nf(n)\floor{n}{i}.
  \]

  整除分块即可。

• \(m=k=4.\)

  显然 \(f_{m,k}(n)=\binom{n}{4}\)。

---

  放三道傻瓜题的原因是让选手享受这题一个情况 \(10\) 分的快乐吗？（

\(\mathcal{Code}\)

/*~Rainybunny~*/

// Have you MODULED correctly?

#ifndef RYBY
#pragma GCC optimize( "Ofast" )
#endif

#include <bits/stdc++.h>

#define rep( i, l, r ) for ( int i = l, rep##i = r; i <= rep##i; ++i )
#define per( i, r, l ) for ( int i = r, per##i = l; i >= per##i; --i )

typedef long long LL;

const int MOD = 1e9 + 7, INV2 = 500000004, INV3 = 333333336, INV4 = 250000002;
LL n;
int m, k;

inline LL sqrs( const LL x ) {
	return x % MOD * ( ( x + 1 ) % MOD ) % MOD * ( ( 2 * x + 1 ) % MOD ) % MOD
	  * INV2 % MOD * INV3 % MOD;
}

inline void solve11() {
	printf( "%lld\n", n % MOD );
}

inline void solve21() {
	LL ans = 0;
	for ( LL l = 1, r; l <= n; l = r + 1 ) {
		r = n / ( n / l );
		ans = ( ans + ( r - l + 1 ) % MOD
		  * ( ( n / l - 1 ) % MOD ) % MOD ) % MOD;
	}
	printf( "%lld\n", ans );
}

inline void solve22() {
	printf( "%lld\n", n % MOD * ( ( n - 1 ) % MOD ) % MOD * INV2 % MOD );
}

inline void solve31() {
	printf( "%lld\n", n / 3 % MOD );
}

inline void solve32() {
	LL ans = 0;
	for ( LL l = 1, r; l <= n; l = r + 1 ) {
		r = n / ( n / l );
		if ( l + ( l & 1 ) <= r - ( r & 1 ) ) { // even.
			LL s = l + ( l & 1 ) - 1 >> 1, t = r - ( r & 1 ) - 1 >> 1;
			LL c = ( r - ( r & 1 ) - ( l + ( l & 1 ) ) >> 1 ) + 1;
			ans = ( ans + n / l % MOD * ( ( s + t ) % MOD ) % MOD
			  * ( c % MOD ) % MOD * INV2 % MOD ) % MOD;
		}
		if ( l + !( l & 1 ) <= r - !( r & 1 ) ) { // odd.
			LL s = l + !( l & 1 ) - 1 >> 1, t = r - !( r & 1 ) - 1 >> 1;
			LL c = ( r - !( r & 1 ) - ( l + !( l & 1 ) ) >> 1 ) + 1;
			ans = ( ans + n / l % MOD * ( ( s + t ) % MOD ) % MOD
			  * ( c % MOD ) % MOD * INV2 % MOD ) % MOD;
		}
	}
	printf( "%lld\n", ans );
}

inline void solve33() {
	printf( "%lld\n", n % MOD * ( ( n - 1 ) % MOD ) % MOD
	  * ( ( n - 2 ) % MOD ) % MOD * INV2 % MOD * INV3 % MOD );
}

inline void solve41() {
	if ( n <= 5 ) return void( puts( "1" ) );
	printf( "%lld\n", ( n / 6 + n / 9 + n / 10
	  + n / 12 + n / 15 + n / 21 ) % MOD );
}

inline void solve42() {
	static const int SMA[] = { 1, 1, 1, 3, 6, 9, 10 };
	if ( n <= 10 ) return void( printf( "%d\n", SMA[n - 4] ) );
	printf( "%lld\n", ( ( n / 11 ) + ( n / 29 ) ) % MOD );
}

inline void solve43() {
	if ( n == 4 ) return void( puts( "1" ) );
	if ( n == 5 ) return void( printf( "%lld\n", n % MOD ) );

	LL ans = 0; // (-MOD,MOD) !!!
	for ( LL l = 6, r; l <= n; l = r + 1 ) {
		r = n / ( n / l );
		LL a = ( sqrs( r - 1 ) - sqrs( l - 2 ) ) % MOD;
		LL b = ( l + r - 2 ) % MOD
		  * ( ( r - l + 1 ) % MOD ) % MOD * INV2 % MOD;
		LL c = 0;
		if ( l + ( l & 1 ) <= r - ( r & 1 ) ) { // even.
			LL s = l + ( l & 1 ) >> 1, t = r - ( r & 1 ) >> 1;
			LL k = ( r - ( r & 1 ) - ( l + ( l & 1 ) ) >> 1 ) + 1;
			c = ( c + ( s + t ) % MOD * ( k % MOD ) % MOD * INV2 % MOD ) % MOD;
		}
		if ( l + !( l & 1 ) <= r - !( r & 1 ) ) { // odd.
			LL s = l + !( l & 1 ) + 1 >> 1, t = r - !( r & 1 ) + 1 >> 1;
			LL k = ( r - !( r & 1 ) - ( l + !( l & 1 ) ) >> 1 ) + 1;
			c = ( c + ( s + t ) % MOD * ( k % MOD ) % MOD * INV2 % MOD ) % MOD;
		}
		ans = ( ans + ( n / l ) % MOD * ( ( a - b ) % MOD * INV2 % MOD
		  - ( 3 * c ) % MOD + 3 * ( r - l + 1 ) % MOD
		  + 2 * ( r / 3 - ( l - 1 ) / 3 ) % MOD ) % MOD
		  * INV2 % MOD * INV3 % MOD ) % MOD;
	}
	printf( "%lld\n", ( ans % MOD + MOD ) % MOD );
}

inline void solve44() {
	printf( "%lld\n", n % MOD * ( ( n - 1 ) % MOD ) % MOD
	  * ( ( n - 2 ) % MOD ) % MOD * ( ( n - 3 ) % MOD ) % MOD
	  * INV2 % MOD * INV3 % MOD * INV4 % MOD );
}

int main() {
	freopen( "vegetable.in", "r", stdin );
	freopen( "vegetable.out", "w", stdout );

	scanf( "%lld %d %d", &n, &m, &k );

	if ( m == 1 ) return solve11(), 0;
	if ( m == 2 ) {
		if ( k == 1 ) return solve21(), 0;
		if ( k == 2 ) return solve22(), 0;
	}
	if ( m == 3 ) {
		if ( k == 1 ) return solve31(), 0;
		if ( k == 2 ) return solve32(), 0;
		if ( k == 3 ) return solve33(), 0;
	}
	if ( m == 4 ) {
		if ( k == 1 ) return solve41(), 0;
		if ( k == 2 ) return solve42(), 0;
		if ( k == 3 ) return solve43(), 0;
		if ( k == 4 ) return solve44(), 0;
	}
	return 0;
}