HDU - 4507 吉哥系列故事――恨7不成妻 (数位DP)

Description

  单身!
  依然单身!
  吉哥依然单身!
  DS级码农吉哥依然单身!
  所以,他生平最恨情人节,不管是214还是77,他都讨厌!
  
  吉哥观察了214和77这两个数,发现:
  2+1+4=7
  7+7=7*2
  77=7*11
  最终,他发现原来这一切归根到底都是因为和7有关!所以,他现在甚至讨厌一切和7有关的数!

  什么样的数和7有关呢?

  如果一个整数符合下面3个条件之一,那么我们就说这个整数和7有关――
  1、整数中某一位是7;
  2、整数的每一位加起来的和是7的整数倍;
  3、这个整数是7的整数倍;

  现在问题来了:吉哥想知道在一定区间内和7无关的数字的平方和。
 

Input

输入数据的第一行是case数T(1 <= T <= 50),然后接下来的T行表示T个case;每个case在一行内包含两个正整数L, R(1 <= L <= R <= 10^18)。
 

Output

请计算[L,R]中和7无关的数字的平方和,并将结果对10^9 + 7 求模后输出。
 

Sample Input

     
     
     
     
3 1 9 10 11 17 17
 

Sample Output

     
     
     
     
236 221 0
 

Source

2013腾讯编程马拉松初赛第一场(3月21日)

思路:数位DP, 一个表示与7无关的个数,一个表示与7无关的数的和,7无关的数字的平方和

           难处理的是平方和,要从前两个推出来,就是将平方和拆出来就行了,还有乱mod不会过啊,

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int maxn = 20;
const ll mod = 1000000007ll;

struct node {
	long long cnt;
	long long sum;
	long long sqsum;
} dp[maxn][10][10];
ll p[maxn];
int num[maxn];

node dfs(int cur, int p1, int p2, int flag) {
	if (cur == -1) {
		node tmp;
		tmp.cnt = (p1 != 0 && p2 != 0);
		tmp.sum = tmp.sqsum = 0;
		return tmp;
	}
	if (!flag && dp[cur][p1][p2].cnt != -1)
		return dp[cur][p1][p2];
	int end = flag?num[cur]:9;
	node ans;
	node tmp;
	ans.cnt = ans.sqsum = ans.sum = 0;
	for (int i = 0; i <= end; i++) {
		if (i == 7)
			continue;
		tmp = dfs(cur-1, (p1+i)%7, (p2*10+i)%7, flag&&i==end);
		ans.cnt += tmp.cnt;
		ans.cnt %= mod;
		ans.sum += (tmp.sum+ ((p[cur]*i)%mod)*tmp.cnt%mod)%mod;
		ans.sum %= mod;
		ans.sqsum += (tmp.sqsum + ((2*p[cur]*i)%mod)*tmp.sum)%mod;
		ans.sqsum %= mod;
		ans.sqsum += (tmp.cnt*p[cur])%mod*p[cur]%mod*i*i%mod;
		ans.sqsum %= mod;
	}
	if (!flag)
		dp[cur][p1][p2] = ans;
	return ans;
}

long long cal(long long m) {
	int cnt = 0;
	while (m) {
		num[cnt++] = m%10;
		m /= 10;
	}
	return dfs(cnt-1, 0, 0, 1).sqsum;
}

int main() {
	int t;
	p[0] = 1;
	for (int i = 1; i < maxn; i++)
		p[i] = (p[i-1]*10) % mod;
	for (int i = 0; i < maxn; i++)
		for (int j = 0; j < 10; j++)
			for (int k = 0; k < 10; k++)
				dp[i][j][k].cnt = -1;
	cin >> t;
	while (t--) {
		ll l, r;
		cin >> l >> r;
		long long ans = cal(r);
		ans -= cal(l-1);
		ans = (ans%mod+mod)%mod;
		cout << ans << endl;
	}
	return 0;
}



你可能感兴趣的:(HDU - 4507 吉哥系列故事――恨7不成妻 (数位DP))