CF550C - Divisibility by Eight

题目链接

https://codeforces.com/problemset/problem/550/C

题目描述

给你一个位数不超过 100 的非负整数 N（不含前导 0）。你的任务是判断这个数字能否通过去掉其中的一些位上的数（当然不能去掉全部），使其成为一个能被 8 整除的正整数（不含前导 0）。特别注意：你不能重新排列数字的顺序。

解题思路

每个数字可以选或不选，像这种需要枚举所有方案数量的题目，大概率可以往DP方面想，当然，还需要判断问题模型最否满足DP的性质（最优子问题和无后效性）。很明显，此题可以DP求解。

在题目描述中，N的位数最多为100，把路径上的全部组合成一个数字再和8取余显然会比较麻烦（高精度），然而我们可以使用同余 的性质，在组合过程中直接和8取余也能得到答案。

本题通过位数划分阶段，每个位数计算时产生的余数可能为0到7，因此每个阶段需要保存至少8个状态的值。

设$dp(i,j)$为前$i$位的基础上进行计算，能否产生余数$j$；
$dp(i,j)=1$ 时表示前$i$位的计算能产生余数$j$；

计算第$i$位能产生那些余数 $m$？

1. 在计算第$i$位时，默认不选择第$i$位，继承$i-1$位的结果即可。

   d p ( i , j ) = d p ( i − 1 , j ) , { 0 < = j < = 7 } dp(i,j)=dp(i-1,j) ,\{0<=j<=7 \} dp(i,j)=dp(i−1,j),{0<=j<=7}

2. 当前位为数的起点时

   $m=A_i\%8;$
   $dp(i,m)=1.$

3. 在第$i-1$位的基础得到结果

   m = ( d p ( i − 1 , j ) ∗ 10 ) % 8 { 0 < = j < = 7 , d p ( i − 1 , j ) > 0 } m= ( dp(i-1,j)*10 )\%8 \{0<=j<=7 ,dp(i-1,j)>0\} m=(dp(i−1,j)∗10)%8{0<=j<=7,dp(i−1,j)>0}
   $dp(i,m)=1.$

在计算第$i$位时如果能产生余数0，直接跳出循环，输出YES。然而题目要求要输出具体数字，在这里我定义了两个数组来保存解。

$mod(i,j)$：第$i$位的算出来的余数$j$由上一位的哪个余数得到的。
$flag(i,j)$：第$i$位上的余数$j$是否为起点。

输出解的思路是：如果mod(i,j)等于mod(i-1,j)，那么第i行的余数是从第i-1行继承过来的，就不是要收集的数字，继续上向走。当遇到flag(i,j)时停止收集数字。

参考代码

代码中还有一些其它不太重要细节就不多说了，大伙自己看吧。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <iomanip>
using namespace std;
const int MAXN = 103; 
char str[MAXN];
int dp[MAXN][8];
int mod[MAXN][8];
bool start_flag[MAXN][8];
char ans[MAXN];
int num = 0;
int find_flag = false;

int main(){
	memset( dp, 0x3f, sizeof(dp) );
	memset( mod, 0x1, sizeof(mod) );

	cin>>str+1;
	
	int t;
	int len = strlen( str+1 );
	for(int i = 1; i <= len; i++){
		//一定要先继承上一行的结果，不能边赋值边计算，不然会出错
		for(int j=0;j<=7;j++){
			dp[i][j] = dp[i-1][j];
			mod[i][j] = mod[i-1][j];
		}
		
		for(int j = 0; j <= 7; j++){
			if( str[i] != '0' ){
				t = (str[i]-'0')%8;
				dp[i][t] = 1;	
				mod[i][t] = t;
				start_flag[i][t] = 1;
			}
			else{
				cout<<"YES\n0"<<endl;
				return 0;
			}		
			if( dp[i-1][j] <= 1 ){
				t = ( j*10+(str[i]-'0') ) % 8;
				dp[i][t] = 1;
				mod[i][t] = j;
			}
		}
		
		if( dp[i][0] == 1 ) {
			
			find_flag=true;	
			cout<<"YES"<<endl;
			ans[ num++] = str[i];
			t = mod[i][0];
			
			if( start_flag[i][t] != 1 )
				for(int j = i-1; j >= 1; j--){
					while( mod[j][t] == mod[j-1][t] ) j--;	//去除余数相等的行 
					ans[num++] = str[j];
					if( start_flag[j][t] == 1 ) break;
					t = mod[j][t];
				}
				
			for(int i = num-1; i >= 0; i-- )
				cout<<ans[i];

			break;
		}
	}
	if( !find_flag ) cout<<"NO"<<endl;

	return 0;
}