제출 #22805

#제출 시각아이디문제언어결과실행 시간메모리
22805버거킹 송죽SK점 우수고객 (#40)Unifying Values (KRIII5_UV)C++14
0 / 7
500 ms2352 KiB
#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);
}

컴파일 시 표준 에러 (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]);
                    ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...