Submission #22805

#	Time	Username	Problem	Language	Result	Execution time	Memory
22805		버거킹 송죽SK점 우수고객 (#40)	Unifying Values (KRIII5_UV)	C++14	0 / 7	500 ms	2352 KiB

UV

#include<bits/stdc++.h>
using namespace std;
const int mod = 1e9+7;

int n, a[10005], s[10005], ans;
int pa[10005], pb[10005], dt[10005];
unordered_map<int, int> mp;

int solve_zero (int idx) {
	mp.clear();
	for(int i=1;i<=n;i++) {
		pa[i] = mp[s[i]-s[idx]];
		pb[i] = mp[s[i]];
		mp[s[i]] = i;
	}
	for(int i=0;i<idx;i++) dt[i] = 0;
	dt[idx] = 1;
	for(int i=idx+1;i<=n;i++) {
		dt[i] = (dt[pa[i]] + dt[pb[i]]) % mod;
	}
	return dt[n];
}

bool valid (int A, int B) {
	return (A % B == 0 && A / B >= 1);
}

int solve (int idx) {
	if(idx == n) return 1;
	if(!valid(s[n], s[idx])) return 0;
	int t = s[n]/s[idx];
	for(int i=0;i<=t;i++) dt[i] = 0;
	dt[1] = 1;
	for(int i=idx+1;i<n;i++) {
		if(!valid(s[i], s[idx])) continue;
		int tt = s[i]/s[idx];
		if(tt >= t) continue;
		dt[tt] = (dt[tt] + dt[tt-1]) % mod;
	}
	return dt[t-1];
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) {
		scanf("%d",&a[i]);
		s[i] = s[i-1] + a[i];
	}
	for(int i=1;i<=n;i++) {
		ans = (ans + (s[i] ? solve(i) : solve_zero(i))) % mod;
	}
	printf("%d\n",(ans+mod-1)%mod);
}

Compilation message (stderr)

UV.cpp: In function 'int main()':
UV.cpp:45:16: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d",&n);
                ^
UV.cpp:47:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d",&a[i]);
                    ^

• Subtask 1: 0 / 2
• Subtask 2: 0 / 5