Submission #22094

#	Time	Username	Problem	Language	Result	Execution time	Memory
22094		sgc109 (#42)	다항식 계산 (KRIII5P_2)	C++98	7 / 7	416 ms	17700 KiB

pc

#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,P;
ll modP[1003][1003]; // [밑][지수]
ll A[1000003];
ll sum[1003];
ll go(int x, int n){
	if(!x) return 0;
	if(n==0) return 1;
	if(n%2){
		ll tmp = go(x,(n-1)/2);
		return (x*tmp*tmp)%P;
	}
	ll tmp = go(x,n/2);
	return (tmp*tmp)%P;
}

int main() {
	inp2(N,P);
	FOR(i,N+1) scanf("%lld",A+i);
	if(P==1){
		printf("%lld",A[N]);
		return 0;
	}
	FOR(i,N+1) sum[i%(P-1)] += A[N-i];
	sum[0] -= A[N];
	FOR(i,P){
		FOR(j,P){
			modP[i][j] = go(i, j);
		}
	}

	FOR(i,P){
		ll ans = 0;
		FOR(j,P){
			ans = (ans + modP[i][j] * sum[j]) % P;
		}
		printf("%lld\n",(ans+A[N])%P);
	}

	return 0;
}

Compilation message (stderr)

pc.cpp: In function 'int main()':
pc.cpp:47:11: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  inp2(N,P);
           ^
pc.cpp:48:30: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  FOR(i,N+1) scanf("%lld",A+i);
                              ^
pc.cpp: At global scope:
pc.cpp:25:14: warning: 'PI' defined but not used [-Wunused-variable]
 const double PI = acos(-1);
              ^

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