Submission #22933

#	Time	Username	Problem	Language	Result	Execution time	Memory
22933		sgc109	Unifying Values (KRIII5_UV)	C++11	7 / 7	306 ms	3120 KiB

UV

#include <bits/stdc++.h>
#define REP(i,a,b) for(int i=a;i<=b;++i)
#define FOR(i,n) for(int i=0;i<n;++i)
#define pb push_back
#define all(v) (v).begin(),(v).end()
#define sz(v) ((int)(v).size())
#define inp1(a) scanf("%d",&a)
#define inp2(a,b) scanf("%d%d",&a,&b)
#define inp3(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define inp4(a,b,c,d) scanf("%d%d%d%d",&a,&b,&c,&d)
#define inp5(a,b,c,d,e) scanf("%d%d%d%d%d",&a,&b,&c,&d,&e)
#define fastio() ios_base::sync_with_stdio(false),cin.tie(NULL)
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pll;
typedef vector<int> vi;	
typedef vector<ll> vl;
typedef pair<int,int> pii;
typedef vector<pii> vii;
typedef vector<pll> vll;
typedef vector<vector<int> > vvi;
typedef pair<int,pair<int,int> > piii;
typedef vector<piii> viii;
const double EPSILON = 1e-9;
const double PI = acos(-1);
const int MOD = 1e9+7;
const int INF = 0x3c3c3c3c;
const long long INFL = 0x3c3c3c3c3c3c3c3c;
const int MAX_N = 102;

int N;
ll dp[10003];
ll pSum[10003];
ll A[10003];
vl valid;

ll go(int pos, ll sum){
	if(pos > N) return 1;
	ll& cache = dp[pos];
	if(cache != -1) return cache;
	cache = 0;
	for(int i = pos; i <= N; i++){
		if(pos==1 && i==N) continue;
		if(pSum[i]-pSum[pos-1] != sum) continue;
		cache = (cache + go(i+1, sum)) % MOD;
	}

	return cache;
}
int main() {
	inp1(N);
	FOR(i,N){
		scanf("%lld",A+i);
		pSum[i+1] = pSum[i] + A[i];
		valid.pb(pSum[i+1]);
	}
	sort(all(valid));
	valid.erase(unique(all(valid)),valid.end());

	ll ans = 0 ;
	FOR(i,sz(valid)){
		if(valid[i] && pSum[N] % valid[i]) continue;
		memset(dp,-1,sizeof(dp));
		ans = (ans + go(1, valid[i])) % MOD;
	}

	printf("%lld",ans);
	return 0;
}

Compilation message (stderr)

UV.cpp: In function 'int main()':
UV.cpp:51:9: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  inp1(N);
         ^
UV.cpp:53:20: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%lld",A+i);
                    ^

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