Torn Lucky Ticket

最新推荐文章于 2025-12-01 22:20:41 发布

原创最新推荐文章于 2025-12-01 22:20:41 发布 · 497 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#算法

codeforces 专栏收录该内容

95 篇文章

订阅专栏

C++代码解决字符串相加问题，运用动态规划，计算不同长度字符串组合的和出现次数。

题目

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 1e6 + 5, d = 500;
vector<string> vec[6];
int mp[1005];
signed main(){
	int n;
	cin >> n;
	for(int i = 1; i <= n; i++){
		string s;
		cin >> s;
		int len = s.size();
		vec[len].push_back(s);
	}
	int res = 0;
	for(int i = 1; i <= 5; i++){
		for(int j = 1; j <= 5; j++){
			if((i + j) % 2 == 1) continue;
			memset(mp, 0, sizeof(mp));
			for(auto s : vec[i]){
				int x = 0;
				for(int k = 0; k < i; k++){
					if(k < (i + j) / 2){
						x += s[k] - '0';
					}
					else{
						x -= s[k] - '0';
					}
				}
				mp[x + d]++;
			}
			for(auto s : vec[j]){
				int x = 0;
				for(int k = 0; k < j; k++){
					if(k < (i + j) / 2){
						x += s[j - 1 - k] - '0';
					}
					else{
						x -= s[j - 1 - k] - '0';
					}
				}
				res += mp[x + d];
			}
		}
	}
	cout << res;
	return 0;
}